Sun, Jimeng

78 publications

FnTML 2025 Artificial Intelligence for Science in Quantum, Atomistic, and Continuum Systems Xuan Zhang, Limei Wang, Jacob Helwig, Youzhi Luo, Cong Fu, Yaochen Xie, Meng Liu, Yuchao Lin, Zhao Xu, Keqiang Yan, Keir Adams, Maurice Weiler, Xiner Li, Tianfan Fu, Yucheng Wang, Alex Strasser, Haiyang Yu, Yuqing Xie, Xiang Fu, Shenglong Xu, Yi Liu, Yuanqi Du, Alexandra Saxton, Hongyi Ling, Hannah Lawrence, Hannes Stärk, Shurui Gui, Carl Edwards, Nicholas Gao, Adriana Ladera, Tailin Wu, Elyssa F. Hofgard, Aria Mansouri Tehrani, Rui Wang, Ameya Daigavane, Montgomery Bohde, Jerry Kurtin, Qian Huang, Tuong Phung, Minkai Xu, Chaitanya K. Joshi, Simon V. Mathis, Kamyar Azizzadenesheli, Ada Fang, Alán Aspuru-Guzik, Erik J. Bekkers, Michael M. Bronstein, Marinka Zitnik, Anima Anandkumar, Stefano Ermon, Pietro Liò, Rose Yu, Stephan Günnemann, Jure Leskovec, Heng Ji, Jimeng Sun, Regina Barzilay, Tommi S. Jaakkola, Connor W. Coley, Xiaoning Qian, Xiaofeng Qian, Tess E. Smidt, Shuiwang Ji

AAAI 2025 Bi-Level Contrastive Learning for Knowledge-Enhanced Molecule Representations Pengcheng Jiang, Cao Xiao, Tianfan Fu, Parminder Bhatia, Taha A. Kass-Hout, Jimeng Sun, Jiawei Han

ICLR 2025 GeSubNet: Gene Interaction Inference for Disease Subtype Network Generation Ziwei Yang, Zheng Chen, Xin Liu, Rikuto Kotoge, Peng Chen, Yasuko Matsubara, Yasushi Sakurai, Jimeng Sun

AAAI 2025 Long-Term EEG Partitioning for Seizure Onset Detection Zheng Chen, Yasuko Matsubara, Yasushi Sakurai, Jimeng Sun

ICLR 2025 Reasoning-Enhanced Healthcare Predictions with Knowledge Graph Community Retrieval Pengcheng Jiang, Cao Xiao, Minhao Jiang, Parminder Bhatia, Taha Kass-Hout, Jimeng Sun, Jiawei Han

NeurIPS 2025 Reinforcement Learning for Out-of-Distribution Reasoning in LLMs: An Empirical Study on Diagnosis-Related Group Coding Hanyin Wang, Zhenbang Wu, Gururaj J. Kolar, Hariprasad Reddy Korsapati, Brian Bartlett, Bryan Hull, Jimeng Sun

ICLR 2025 Small Models Are LLM Knowledge Triggers for Medical Tabular Prediction Jiahuan Yan, Jintai Chen, Chaowen Hu, Bo Zheng, Yaojun Hu, Jimeng Sun, Jian Wu

NeurIPS 2024 CARES: A Comprehensive Benchmark of Trustworthiness in Medical Vision Language Models Peng Xia, Ze Chen, Juanxi Tian, Yangrui Gong, Ruibo Hou, Yue Xu, Zhenbang Wu, Zhiyuan Fan, Yiyang Zhou, Kangyu Zhu, Wenhao Zheng, Zhaoyang Wang, Xiao Wang, Xuchao Zhang, Chetan Bansal, Marc Niethammer, Junzhou Huang, Hongtu Zhu, Yun Li, Jimeng Sun, Zongyuan Ge, Gang Li, James Zou, Huaxiu Yao

ICMLW 2024 CARES: A Comprehensive Benchmark of Trustworthiness in Medical Vision Language Models Peng Xia, Ze Chen, Juanxi Tian, Gong Yangrui, Ruibo Hou, Yue Xu, Zhenbang Wu, Zhiyuan Fan, Yiyang Zhou, Kangyu Zhu, Wenhao Zheng, Zhaoyang Wang, Xiao Wang, Xuchao Zhang, Chetan Bansal, Marc Niethammer, Junzhou Huang, Hongtu Zhu, Yun Li, Jimeng Sun, Zongyuan Ge, Gang Li, James Zou, Huaxiu Yao

ICML 2024 Certifiably Byzantine-Robust Federated Conformal Prediction Mintong Kang, Zhen Lin, Jimeng Sun, Cao Xiao, Bo Li

AAAI 2024 ConSequence: Synthesizing Logically Constrained Sequences for Electronic Health Record Generation Brandon Theodorou, Shrusti Jain, Cao Xiao, Jimeng Sun

TMLR 2024 Generating with Confidence: Uncertainty Quantification for Black-Box Large Language Models Zhen Lin, Shubhendu Trivedi, Jimeng Sun

ICLR 2024 GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge Graphs Pengcheng Jiang, Cao Xiao, Adam Richard Cross, Jimeng Sun

NeurIPS 2024 Instruction Tuning Large Language Models to Understand Electronic Health Records Zhenbang Wu, Anant Dadu, Mike Nalls, Faraz Faghri, Jimeng Sun

NeurIPS 2024 KG-FIT: Knowledge Graph Fine-Tuning upon Open-World Knowledge Pengcheng Jiang, Lang Cao, Cao Xiao, Parminder Bhatia, Jimeng Sun, Jiawei Han

ICLR 2024 Making Pre-Trained Language Models Great on Tabular Prediction Jiahuan Yan, Bo Zheng, Hongxia Xu, Yiheng Zhu, Danny Chen, Jimeng Sun, Jian Wu, Jintai Chen

IJCAI 2024 MediTab: Scaling Medical Tabular Data Predictors via Data Consolidation, Enrichment, and Refinement Zifeng Wang, Chufan Gao, Cao Xiao, Jimeng Sun

ICLR 2024 Multimodal Patient Representation Learning with Missing Modalities and Labels Zhenbang Wu, Anant Dadu, Nicholas Tustison, Brian Avants, Mike Nalls, Jimeng Sun, Faraz Faghri

IJCAI 2024 Recent Advances in Predictive Modeling with Electronic Health Records Jiaqi Wang, Junyu Luo, Muchao Ye, Xiaochen Wang, Yuan Zhong, Aofei Chang, Guanjie Huang, Ziyi Yin, Cao Xiao, Jimeng Sun, Fenglong Ma

NeurIPSW 2024 TrialSynth: Generation of Synthetic Sequential Clinical Trial Data Chufan Gao, Mandis Beigi, Afrah Shafquat, Jacob Aptekar, Jimeng Sun

ICMLW 2023 An Interpretable Data Augmentation Framework for Improving Generative Modeling of Synthetic Clinical Trial Data Afrah Shafquat, Jason Mezey, Mandis Beigi, Jimeng Sun, Andy Gao, Jacob W. Aptekar

NeurIPS 2023 An Iterative Self-Learning Framework for Medical Domain Generalization Zhenbang Wu, Huaxiu Yao, David Liebovitz, Jimeng Sun

NeurIPS 2023 BIOT: Biosignal Transformer for Cross-Data Learning in the Wild Chaoqi Yang, M Westover, Jimeng Sun

NeurIPS 2023 CoDrug: Conformal Drug Property Prediction with Density Estimation Under Covariate Shift Siddhartha Laghuvarapu, Zhen Lin, Jimeng Sun

ICML 2023 Fast Online Value-Maximizing Prediction Sets with Conformal Cost Control Zhen Lin, Shubhendu Trivedi, Cao Xiao, Jimeng Sun

ICLR 2023 ManyDG: Many-Domain Generalization for Healthcare Applications Chaoqi Yang, M Brandon Westover, Jimeng Sun

NeurIPSW 2023 Stoichiometry Representation Learning with Polymorphic Crystal Structures Namkyeong Lee, Heewoong Noh, Gyoung S. Na, Tianfan Fu, Jimeng Sun, Chanyoung Park

ICLR 2023 Taking a Step Back with KCal: Multi-Class Kernel-Based Calibration for Deep Neural Networks Zhen Lin, Shubhendu Trivedi, Jimeng Sun

NeurIPSW 2022 A Source Data Privacy Framework for Synthetic Clinical Trial Data Afrah Shafquat, Jason Mezey, Mandis Beigi, Jimeng Sun, Jacob W. Aptekar

NeurIPS 2022 ATD: Augmenting CP Tensor Decomposition by Self Supervision Chaoqi Yang, Cheng Qian, Navjot Singh, Cao Xiao, M Westover, Edgar Solomonik, Jimeng Sun

ECML-PKDD 2022 AutoMap: Automatic Medical Code Mapping for Clinical Prediction Model Deployment Zhenbang Wu, Cao Xiao, Lucas M. Glass, David M. Liebovitz, Jimeng Sun

ECCV 2022 CT2: Colorization Transformer via Color Tokens Shuchen Weng, Jimeng Sun, Yu Li, Si Li, Boxin Shi

TMLR 2022 Conformal Prediction Intervals with Temporal Dependence Zhen Lin, Shubhendu Trivedi, Jimeng Sun

NeurIPS 2022 Conformal Prediction with Temporal Quantile Adjustments Zhen Lin, Shubhendu Trivedi, Jimeng Sun

ICLR 2022 Differentiable Scaffolding Tree for Molecule Optimization Tianfan Fu, Wenhao Gao, Cao Xiao, Jacob Yasonik, Connor W. Coley, Jimeng Sun

MLHC 2022 EMIXER: End-to-End Multimodal X-Ray Generation via Self-Supervision Siddharth Biswal, Peiye Zhuang, Ayis Pyrros, Nasir Siddiqui, Sanmi Koyejo, Jimeng Sun

IJCAI 2022 GOCPT: Generalized Online Canonical Polyadic Tensor Factorization and Completion Chaoqi Yang, Cheng Qian, Jimeng Sun

ICLR 2022 PAC-Bayes Information Bottleneck Zifeng Wang, Shao-Lun Huang, Ercan Engin Kuruoglu, Jimeng Sun, Xi Chen, Yefeng Zheng

NeurIPSW 2022 Recommendation for New Drugs with Limited Prescription Data Zhenbang Wu, Huaxiu Yao, Zhe Su, David Liebovitz, Lucas M Glass, James Zou, Chelsea Finn, Jimeng Sun

NeurIPS 2022 Reinforced Genetic Algorithm for Structure-Based Drug Design Tianfan Fu, Wenhao Gao, Connor Coley, Jimeng Sun

ICMLW 2022 Reinforced Genetic Algorithm for Structure-Based Drug Design Tianfan Fu, Wenhao Gao, Connor W. Coley, Jimeng Sun

AAAI 2022 SCRIB: Set-Classifier with Class-Specific Risk Bounds for Blackbox Models Zhen Lin, Lucas Glass, M. Brandon Westover, Cao Xiao, Jimeng Sun

NeurIPS 2022 Sample Efficiency Matters: A Benchmark for Practical Molecular Optimization Wenhao Gao, Tianfan Fu, Jimeng Sun, Connor Coley

ICMLW 2022 Sample Efficiency Matters: A Benchmark for Practical Molecular Optimization Wenhao Gao, Tianfan Fu, Jimeng Sun, Connor W. Coley

NeurIPS 2022 TransTab: Learning Transferable Tabular Transformers Across Tables Zifeng Wang, Jimeng Sun

CVPR 2022 UniCoRN: A Unified Conditional Image Repainting Network Jimeng Sun, Shuchen Weng, Zheng Chang, Si Li, Boxin Shi

IJCAI 2021 Change Matters: Medication Change Prediction with Recurrent Residual Networks Chaoqi Yang, Cao Xiao, Lucas Glass, Jimeng Sun

MLHC 2021 EVA: Generating Longitudinal Electronic Health Records Using Conditional Variational Autoencoders Siddharth Biswal, Soumya Ghosh, Jon Duke, Bradley Malin, Walter Stewart, Cao Xiao, Jimeng Sun

NeurIPS 2021 Locally Valid and Discriminative Prediction Intervals for Deep Learning Models Zhen Lin, Shubhendu Trivedi, Jimeng Sun

AAAI 2021 MIMOSA: Multi-Constraint Molecule Sampling for Molecule Optimization Tianfan Fu, Cao Xiao, Xinhao Li, Lucas M. Glass, Jimeng Sun

IJCAI 2021 Multi-Version Tensor Completion for Time-Delayed Spatio-Temporal Data Cheng Qian, Nikos Kargas, Cao Xiao, Lucas Glass, Nicholas D. Sidiropoulos, Jimeng Sun

AAAI 2021 STELAR: Spatio-Temporal Tensor Factorization with Latent Epidemiological Regularization Nikos Kargas, Cheng Qian, Nicholas D. Sidiropoulos, Cao Xiao, Lucas M. Glass, Jimeng Sun

AAAI 2021 SWIFT: Scalable Wasserstein Factorization for Sparse Nonnegative Tensors Ardavan Afshar, Kejing Yin, Sherry Yan, Cheng Qian, Joyce C. Ho, Haesun Park, Jimeng Sun

IJCAI 2021 SafeDrug: Dual Molecular Graph Encoders for Recommending Effective and Safe Drug Combinations Chaoqi Yang, Cao Xiao, Fenglong Ma, Lucas Glass, Jimeng Sun

AAAI 2020 CASTER: Predicting Drug Interactions with Chemical Substructure Representation Kexin Huang, Cao Xiao, Trong Nghia Hoang, Lucas Glass, Jimeng Sun

AAAI 2020 CONAN: Complementary Pattern Augmentation for Rare Disease Detection Limeng Cui, Siddharth Biswal, Lucas M. Glass, Greg Lever, Jimeng Sun, Cao Xiao

AAAI 2020 CORE: Automatic Molecule Optimization Using Copy & Refine Strategy Tianfan Fu, Cao Xiao, Jimeng Sun

AAAI 2020 Doctor2Vec: Dynamic Doctor Representation Learning for Clinical Trial Recruitment Siddharth Biswal, Cao Xiao, Lucas M. Glass, Elizabeth Milkovits, Jimeng Sun

ICML 2020 SDE-Net: Equipping Deep Neural Networks with Uncertainty Estimates Lingkai Kong, Jimeng Sun, Chao Zhang

IJCAI 2019 DDL: Deep Dictionary Learning for Predictive Phenotyping Tianfan Fu, Trong Nghia Hoang, Cao Xiao, Jimeng Sun

MLHC 2019 EEGtoText: Learning to Write Medical Reports from EEG Recordings Siddharth Biswal, Cao Xiao, M. Brandon Westover, Jimeng Sun

AAAI 2019 GAMENet: Graph Augmented MEmory Networks for Recommending Medication Combination Junyuan Shang, Cao Xiao, Tengfei Ma, Hongyan Li, Jimeng Sun

UAI 2019 Guaranteed Scalable Learning of Latent Tree Models Furong Huang, Niranjan Uma Naresh, Ioakeim Perros, Robert Chen, Jimeng Sun, Anima Anandkumar

AAAI 2019 Hierarchical Reinforcement Learning for Course Recommendation in MOOCs Jing Zhang, Bowen Hao, Bo Chen, Cuiping Li, Hong Chen, Jimeng Sun

IJCAI 2019 MINA: Multilevel Knowledge-Guided Attention for Modeling Electrocardiography Signals Shenda Hong, Cao Xiao, Tengfei Ma, Hongyan Li, Jimeng Sun

IJCAI 2019 Pre-Training of Graph Augmented Transformers for Medication Recommendation Junyuan Shang, Tengfei Ma, Cao Xiao, Jimeng Sun

IJCAI 2019 RDPD: Rich Data Helps Poor Data via Imitation Shenda Hong, Cao Xiao, Trong Nghia Hoang, Tengfei Ma, Hongyan Li, Jimeng Sun

MLHC 2019 SLEEPER: Interpretable Sleep Staging via Prototypes from Expert Rules Irfan Al-Hussaini, Cao Xiao, M. Brandon Westover, Jimeng Sun

NeurIPS 2018 MiME: Multilevel Medical Embedding of Electronic Health Records for Predictive Healthcare Edward Choi, Cao Xiao, Walter Stewart, Jimeng Sun

MLHC 2017 Generating Multi-Label Discrete Patient Records Using Generative Adversarial Networks Edward Choi, Siddharth Biswal, Bradley Malin, Jon Duke, Walter F. Stewart, Jimeng Sun

AAAI 2017 StructInf: Mining Structural Influence from Social Streams Jing Zhang, Jie Tang, Yuanyi Zhong, Yuchen Mo, Juanzi Li, Guojie Song, Wendy Hall, Jimeng Sun

MLHC 2016 Doctor AI: Predicting Clinical Events via Recurrent Neural Networks Edward Choi, Mohammad Taha Bahadori, Andy Schuetz, Walter F. Stewart, Jimeng Sun

NeurIPS 2016 RETAIN: An Interpretable Predictive Model for Healthcare Using Reverse Time Attention Mechanism Edward Choi, Mohammad Taha Bahadori, Jimeng Sun, Joshua Kulas, Andy Schuetz, Walter Stewart

NeurIPS 2015 Time-Sensitive Recommendation from Recurrent User Activities Nan Du, Yichen Wang, Niao He, Jimeng Sun, Le Song

AAAI 2011 Automatic Group Sparse Coding Fei Wang, Noah Lee, Jimeng Sun, Jianying Hu, Shahram Ebadollahi

ECML-PKDD 2008 Hierarchical, Parameter-Free Community Discovery Spiros Papadimitriou, Jimeng Sun, Christos Faloutsos, Philip S. Yu

ECML-PKDD 2008 Two Heads Better than One: Pattern Discovery in Time-Evolving Multi-Aspect Data Jimeng Sun, Charalampos E. Tsourakakis, Evan Hoke, Christos Faloutsos, Tina Eliassi-Rad

IJCAI 2007 Dynamic Mixture Models for Multiple Time-Series Xing Wei, Jimeng Sun, Xuerui Wang