Figure 1.

Overview of scPRAM. scPRAM employs the encoder of a VAE to map the cell-gene expression matrix into a latent space. Then, optimal transport is utilized to compute the coupling matrix between cells before and after perturbation, facilitating the matching of each unperturbed cell with its corresponding perturbed cell. Considering the latent vector of the test cell as a query, the latent space of the matched training set as a corpus, scPRAM employs attention mechanism to calculate the perturbation vector corresponding to each query. The perturbation vectors are added to their respective queries, and the predicted perturbation response is obtained through the output of the decoder.