Spatial transcriptomics stops swapping only cell-level states: SpaAxFormer keeps each gene's state and beats Impeller on DLPFC imputation L1
Spatial transcriptomics models usually exchange representations only between cells or spots, with gene information arriving through the attached expression profile. SpaAxFormer keeps every cell–gene entry as its own state, first letting genes within a cell reference each other, then exchanging the same gene's state across spatial neighbors, with a separate cell-level summary. On six DLPFC slices its mean L1 imputation distance was lower than Impeller's, for example 0.239 versus 0.254 on slice 151510; on Stereo-seq and Slide-seq V2, Impeller had lower mean L1 and RMSE.