Proceedings

URFPert: Unrolled Regulatory Flow Networks for Out-of-Distribution Single-Cell Perturbation Prediction

Download BibTeX Cite

Proceedings of the Thirty-Fifth International Joint Conference on Artificial Intelligence. Main Track. Pages 4795-4803.

https://doi.org/10.24963/ijcai.2026/534

Abstract

Single-cell perturbation screens enable systematic discovery of gene regulatory mechanisms, yet the exponential expansion of perturbation space makes comprehensive experimentation impractical. Although in silico predictors have been increasingly proposed to address this challenge, most existing methods either assume static regulatory priors or learn unconstrained data-driven representations. As a result, they often suffer from poor adaptation to state-dependent shifts and weak out-of-distribution generalization. In response, we propose URFPert, an unrolled regulatory flow network tailored for single-cell perturbation prediction. Specifically, URFPert is composed of three fundamental modules: (i) Regulatory Structure Unrolling module, which employs an iterative unrolling optimization algorithm to infer state-dependent gene regulatory networks (GRNs); (ii) Heterogeneous Graph Flow Network, which incorporates the inferred GRNs into a conditional velocity field to yield state-consistent and perturbation-specific trajectories; (iii)Discriminative Optimization Strategy, which combines Mean Squared Error~(MSE) with Maximal Coding Rate Reduction~(MCR²) to promote discriminative perturbation responses and mitigate training confounders. Comprehensive evaluations on five single-cell perturbation datasets demonstrate that URFPert outperforms state-of-the-art methods in unseen perturbation regimes, providing a powerful tool for interpreting regulatory mechanisms. The project code is public via https://github.com/Jiniretttt/URFPert.

  1. [Ahlmann-Eltze et al., 2025] Constantin Ahlmann-Eltze, Wolfgang Huber, and Simon Anders. Deep-learning- based gene perturbation effect prediction does not yet outperform simple linear baselines. Nature Methods, 22(8):1657-1661, 2025.
  2. [Barry et al., 2021] Timothy Barry, Xuran Wang, John A Morris, Kathryn Roeder, and Eugene Katsevich. Scep- tre improves calibration and sensitivity in single-cell crispr screen analysis. Genome biology, 22(1):344, 2021.
  3. [Beck and Teboulle, 2009] Amir Beck and Marc Teboulle. A fast iterative shrinkage-thresholding algorithm for lin- ear inverse problems. SIAM journal on imaging sciences, 2(1):183-202, 2009.
  4. [Bunne et al., 2023] Charlotte Bunne, Stefan G Stark, Gabriele Gut, Jacobo Sarabia Del Castillo, Mitch Levesque, Kjong-Van Lehmann, Lucas Pelkmans, An- dreas Krause, and Gunnar Rätsch. Learning single-cell perturbation responses using neural optimal transport. Na- ture methods, 20(11):1759-1768, 2023.
  5. [Busbridge et al., 2019] Dan Busbridge, Dane Sherburn, Pietro Cavallo, and Nils Y Hammerla. Relational graph at- tention networks. arXiv preprint arXiv:1904.05811, 2019.
  6. [Chen and Zou, 2025] Yiqun Chen and James Zou. Sim- ple and effective embedding model for single-cell biol- ogy built from chatgpt. Nature biomedical engineering, 9(4):483-493, 2025.
  7. [Chen et al., 2018] Ricky TQ Chen, Yulia Rubanova, Jesse Bettencourt, and David K Duvenaud. Neural ordinary dif- ferential equations. Advances in neural information pro- cessing systems, 31, 2018.
  8. [Consortium, 2021] The Gene Ontology Consortium. The gene ontology resource: enriching a gold mine. Nucleic Acids Research, 49(D1):D325-D334, 01 2021.
  9. [Cui et al., 2024] Haotian Cui, Chloe Wang, Hassaan Maan, Kuan Pang, Fengning Luo, Nan Duan, and Bo Wang. scgpt: toward building a foundation model for single- cell multi-omics using generative ai. Nature methods, 21(8):1470-1480, 2024.
  10. [Dhar et al., 2013] Manoj K Dhar, Archana Koul, and San- jana Kaul. Farnesyl pyrophosphate synthase: a key en- zyme in isoprenoid biosynthetic pathway and potential molecular target for drug development. New biotechnol- ogy, 30(2):114-123, 2013.
  11. [Dixit et al., 2016] Atray Dixit, Oren Parnas, Biyu Li, Jenny Chen, Charles P Fulco, Livnat Jerby-Arnon, Nemanja D Marjanovic, Danielle Dionne, Tyler Burks, Raktima Ray- chowdhury, et al. Perturb-seq: dissecting molecular cir- cuits with scalable single-cell rna profiling of pooled ge- netic screens. cell, 167(7):1853-1866, 2016.
  12. [Esser et al., 2024] Patrick Esser, Sumith Kulal, Andreas Blattmann, Rahim Entezari, Jonas Müller, Harry Saini, Yam Levi, Dominik Lorenz, Axel Sauer, Frederic Boe- sel, et al. Scaling rectified flow transformers for high- resolution image synthesis. In Forty-first international conference on machine learning, 2024.
  13. [Goeva et al., 2024] Aleksandrina Goeva, Michael-John Dolan, Judy Luu, Eric Garcia, Rebecca Boiarsky, Rajat M Gupta, and Evan Macosko. Hidden: a machine learning method for detection of disease-relevant populations in case-control single-cell transcriptomics data. Nature Communications, 15(1):9468, 2024.
  14. [Liu et al., 2022] Xingchao Liu, Chengyue Gong, and Qiang Liu. Flow straight and fast: Learning to generate and transfer data with rectified flow. arXiv preprint arXiv:2209.03003, 2022.
  15. [Loshchilov and Hutter, 2017] Ilya Loshchilov and Frank Hutter. Decoupled weight decay regularization. arXiv preprint arXiv:1711.05101, 2017.
  16. [Lotfollahi et al., 2023] Mohammad Lotfollahi, Anna Klimovskaia Susmelj, Carlo De Donno, Leon Hetzel, Yuge Ji, Ignacio L Ibarra, Sanjay R Srivatsan, Mohsen Naghipourfar, Riza M Daza, Beth Martin, et al. Pre- dicting cellular responses to complex perturbations in high-throughput screens. Molecular systems biology, 19(6):e11517, 2023.
  17. [Macek et al., 2018] Pavel Macek, Matthew J Cliff, Kevin J Embrey, Geoffrey A Holdgate, J Willem M Nissink, Stanislava Panova, Jonathan P Waltho, and Rick A Davies. Myc phosphorylation in its basic helix-loop-helix re- gion destabilizes transient α-helical structures, disrupting max and dna binding. Journal of Biological Chemistry, 293(24):9301-9310, 2018.
  18. [Monga et al., 2021] Vishal Monga, Yuelong Li, and Yon- ina C Eldar. Algorithm unrolling: Interpretable, efficient deep learning for signal and image processing. IEEE Sig- nal Processing Magazine, 38(2):18-44, 2021.
  19. [Pakos-Zebrucka et al., 2016] Karolina Pakos-Zebrucka, Iz- abela Koryga, Katarzyna Mnich, Mila Ljujic, Afshin Samali, and Adrienne M Gorman. The integrated stress response. EMBO reports, 17(10):1374-1395, 2016.
  20. [Papalexi et al., 2021] Efthymia Papalexi, Eleni P Mimitou, Andrew W Butler, Samantha Foster, Bernadette Bracken, William M Mauck III, Hans-Hermann Wessels, Yuhan Hao, Bertrand Z Yeung, Peter Smibert, et al. Characteriz- ing the molecular regulation of inhibitory immune check- points with multimodal single-cell screens. Nature genet- ics, 53(3):322-331, 2021.
  21. [Replogle et al., 2022] Joseph M Replogle, Reuben A Saun- ders, Angela N Pogson, Jeffrey A Hussmann, Alexan- der Lenail, Alina Guna, Lauren Mascibroda, Eric J Wag- ner, Karen Adelman, Gila Lithwick-Yanai, et al. Map- ping information-rich genotype-phenotype landscapes with genome-scale perturb-seq. Cell, 185(14):2559-2575, 2022.
  22. [Roohani et al., 2024] Yusuf Roohani, Kexin Huang, and Jure Leskovec. Predicting transcriptional outcomes of novel multigene perturbations with gears. Nature Biotech- nology, 42(6):927-935, 2024.
  23. [Roohani et al., 2025] Yusuf H Roohani, Tony J Hua, Po- Yuan Tung, Lexi R Bounds, Feiqiao B Yu, Alexander Dobin, Noam Teyssier, Abhinav Adduri, Alden Woodrow, Brian S Plosky, et al. Virtual cell challenge: Toward a turing test for the virtual cell. Cell, 188(13):3370-3374, 2025.
  24. [Saboksayr and Mateos, 2021] Seyed Saman Saboksayr and Gonzalo Mateos. Accelerated graph learning from smooth signals. IEEE Signal Processing Letters, 28:2192-2196, 2021.
  25. [Tancik et al., 2020] Matthew Tancik, Pratul Srinivasan, Ben Mildenhall, Sara Fridovich-Keil, Nithin Raghavan, Utkarsh Singhal, Ravi Ramamoorthi, Jonathan Barron, and Ren Ng. Fourier features let networks learn high fre- quency functions in low dimensional domains. Advances in neural information processing systems, 33:7537-7547, 2020.
  26. [Tian et al., 2021] Ruilin Tian, Anthony Abarientos, Jason Hong, Sayed Hadi Hashemi, Rui Yan, Nina Dräger, Kun Leng, Mike A Nalls, Andrew B Singleton, Ke Xu, et al. Genome-wide crispri/a screens in human neurons link lysosomal failure to ferroptosis. Nature neuroscience, 24(7):1020-1034, 2021.
  27. [Van de Sande et al., 2020] Bram Van de Sande, Christopher Flerin, Kristofer Davie, Maxime De Waegeneer, Gert Hulselmans, Sara Aibar, Ruth Seurinck, Wouter Saelens, Robrecht Cannoodt, Quentin Rouchon, et al. A scalable scenic workflow for single-cell gene regulatory network analysis. Nature protocols, 15(7):2247-2276, 2020.
  28. [Wan et al., 2011] Guohui Wan, Rohit Mathur, Xiaoxiao Hu, Xinna Zhang, and Xiongbin Lu. mirna response to dna damage. Trends in biochemical sciences, 36(9):478-484, 2011.
  29. [Wei et al., 2025] Zhiting Wei, Yiheng Wang, Yicheng Gao, Shuguang Wang, Ping Li, Duanmiao Si, Yuli Gao, Siqi Wu, Danlu Li, Kejing Dong, et al. Benchmarking algo- rithms for generalizable single-cell perturbation response prediction. Nature Methods, pages 1-14, 2025.
  30. [Wessels et al., 2023] Hans-Hermann Wessels, Alejan- dro Méndez-Mancilla, Yuhan Hao, Efthymia Papalexi, William M Mauck III, Lu Lu, John A Morris, Eleni P Mimitou, Peter Smibert, Neville E Sanjana, et al. Efficient combinatorial targeting of rna transcripts in single cells with cas13 rna perturb-seq. Nature methods, 20(1):86-94, 2023.
  31. [Yu et al., 2020] Yaodong Yu, Kwan Ho Ryan Chan, Chong You, Chaobing Song, and Yi Ma. Learning diverse and discriminative representations via the principle of maxi- mal coding rate reduction. Advances in neural information processing systems, 33:9422-9434, 2020.
  32. [Zhu and Li, 2025] Ouyang Zhu and Jun Li. Scouter pre- dicts transcriptional responses to genetic perturbations with large language model embeddings. Nature Compu- tational Science, pages 1-8, 2025.
  33. [Zou et al., 2018] Jiahuan Zou, Mengwei Zhuang, Xiaopeng Yu, Na Li, Rudi Mao, Zhida Wang, Jianing Wang, Xiaoyan Wang, Huaiyu Zhou, Lining Zhang, et al. Myc inhibition increases pd-l1 expression induced by ifn-γ in hepatocel- lular carcinoma cells. Molecular immunology, 101:203- 209, 2018.

References were extracted automatically from the PDF and may contain errors.

Welcome Future Conferences AI Journal Division IJCAI Proceedings About IJCAI IJCAI Awards