Keiji AI LogoKeiji AI

Research Publications

Browse papers by research area or filter by keyword.

Methods for designing more feasible trials, simulating outcomes, and creating analysis-ready evidence for statistical teams.

Trial design and simulationStatistical programmingDigital twins and synthetic controls
See solution use cases
EMNLP'23

AutoTrial: Prompting Language Models for Clinical Trial Design

Zifeng Wang, Cao Xiao, Jimeng Sun

Read Paper
EMNLP'22

Trial2Vec: Zero-Shot Clinical Trial Document Similarity Search using Self-Supervision

Zifeng Wang, Jimeng Sun

Read Paper
KDD'23

TWIN: Personalized Clinical Trial Digital Twin Generation

Trisha Das, Zifeng Wang, Jimeng Sun

Read Paper
Patterns

Clinical trial outcome prediction with multi-modal generative AI

Brandon Theodorou, Zifeng Wang, Jimeng Sun

Read Paper

Reliable methods for cohort analysis, longitudinal patient modeling, comparative effectiveness, and safety studies on real-world data.

Cohort feasibilityComparative effectivenessTreatment patterns and safety
See solution use cases
Preprint

UniPredict: Large Language Models are Universal Tabular Predictors

Ruiyu Wang, Zifeng Wang, Jimeng Sun

Read Paper
IJCAI'24

MediTab: Scaling Medical Tabular Data Predictors via Data Consolidation, Enrichment, and Refinement

Zifeng Wang, Chufan Gao,Cao Xiao, Jimeng Sun

Read Paper
NeurIPS'22

TransTab: Learning Transferable Tabular Transformers Across Tables

Zifeng Wang, Jimeng Sun

Read Paper
JAMIA'21

STAN: Spatio-Temporal Attention Network for Pandemic Prediction using Real-World Evidence

Junyi Gao, Rakshith Sharma, Cheng Qian, Lucas M Glass, Jeffrey Spaeder, Justin Romberg, Jimeng Sun, Cao Xiao

Read Paper
Nature Comm

Evidence-driven spatiotemporal COVID-19 hospitalization prediction with Ising dynamics

Junyi Gao, Joerg Heintz, Christina Mack, Lucas Glass, Adam Cross & Jimeng Sun

Read Paper
WWW'22

PopNet: Real-Time Population-Level Disease Prediction with Data Latency

Junyi Gao, Cao Xiao, Lucas M. Glass, Jimeng Sun

Read Paper
npj Digital Med

Improving medical machine learning models with generative balancing for equity and excellence

Brandon Theodorou, Benjamin Danek, Venkat Tummala, Shivam Pankaj Kumar, Bradley Malin, Jimeng Sun

Read Paper
Nature Comm

Synthesize high-dimensional longitudinal electronic health records via hierarchical autoregressive language model

Brandon Theodorou, Cao Xiao, Jimeng Sun

Read Paper
EMNLP'22

PromptEHR: Conditional Electronic Healthcare Records Generation with Prompt Learning

Zifeng Wang, Jimeng Sun

Read Paper

Clinical-trial foundation models and predictive methods for landscape analysis, probability of success, and evidence-backed portfolio decisions.

Trial landscape and KOL discoveryProbability of technical successPortfolio and indication assessment
See solution use cases
npj Digital Med

Benchmarking and Developing Large Language Models Using One Million Clinical Trials (TrialPanorama)

Zifeng Wang, Jiacheng Lin, Qiao Jin, Junyi Gao, Jathurshan Pradeepkumar, Pengcheng Jiang, Zhiyong Lu, Jimeng Sun

Read Paper
Preprint

Panacea: A foundation model for clinical trial search, summarization, design, and recruitment

Jiacheng Lin, Hanwen Xu, Zifeng Wang, Sheng Wang, Jimeng Sun

Read Paper
ACM-BCB'23

SPOT: Sequential Predictive Modeling of Clinical Trial Outcome with Meta-Learning

Zifeng Wang, Jimeng Sun

Read Paper
Patterns

HINT: Hierarchical Interaction Network for Clinical Trial Outcome Predictions

Zifeng Wang, Jimeng Sun

Read Paper
Scientific Data

TrialBench: Multi-Modal AI-Ready Datasets for Clinical Trial Prediction

Jintai Chen, Yaojun Hu, Mingchen Cai, Yingzhou Lu, Yue Wang, Xu Cao, Miao Lin, Hongxia Xu, Jian Wu, Xiao Cao, Jimeng Sun, Yuqiang Li, Lucas Glass, Kexin Huang, Marinka Zitnik, Tianfan Fu

Read Paper

Research that improves site selection, patient-trial matching, recruitment, and the operational decisions that determine whether a trial finishes.

Evidence-based site selectionPatient matching and recruitmentEnrollment and dropout prediction
See solution use cases
Patterns

FRAMM: Fair ranking with missing modalities for clinical trial site selection

Brandon Theodorou, Lucas Glass, Cao Xiao, Jimeng Sun

Read Paper
Nature Comm

Matching patients to clinical trials with large language models

Qiao Jin, Zifeng Wang, Charalampos S. Floudas, Fangyuan Chen, Changlin Gong, Dara Bracken-Clarke, Elisabetta Xue, Yifan Yang, Jimeng Sun & Zhiyong Lu

Read Paper
KDD'20

COMPOSE: Cross-modal pseudo-siamese network for patient trial matching

Junyi Gao, Cao Xiao, Lucas M. Glass, Jimeng Sun

Read Paper
AAAI'20

Doctor2Vec: Dynamic Doctor Representation Learning for Clinical Trial Recruitment

Junyi Gao, Cao Xiao, Lucas M. Glass, Jimeng Sun

Read Paper

Human-AI evidence synthesis methods for systematic reviews, reproducible screening, and evidence bases that can stand up to HTA scrutiny.

Systematic literature reviewPRISMA screening and extractionReusable HTA evidence bases
See solution use cases
Nature Mach Intell

Empowering Biomedical Evidence Exploration and Synthesis with Deep Knowledge Graph Research

Zifeng Wang, Zheng Chen, Ziwei Yang, Xiao Wang, Qiao Jin, Yifan Peng, Zhiyong Lu, Jimeng Sun

Read Paper
npj Digital Med

Accelerating clinical evidence synthesis with large language models

Wang, Zifeng and Cao, Lang and Danek, Benjamin and Jin, Qiao and Lu, Zhiyong and Sun, Jimeng

Read Paper
Nature Comm

A foundation model for human-AI collaboration in medical literature mining

Zifeng Wang, Lang Cao, Qiao Jin, Joey Chan, Nicholas Wan, Behdad Afzali, Hyun-Jin Cho, Chang-In Choi, Mehdi Emamverdi, Manjot K. Gill, Sun-Hyung Kim, Yijia Li, Yi Liu, Hanley Ong, Justin Rousseau, Irfan Sheikh, Jenny J. Wei, Ziyang Xu, Christopher M. Zallek, Kyungsang Kim, Yifan Peng, Zhiyong Lu, Jimeng Sun

Read Paper

Foundational AI, including drug-discovery and multimodal methods, plus data-science agents that help investigators run reproducible biomedical studies.

Investigator-led data scienceMultimodal biomedical researchReproducible methods and training
See solution use cases
Nature BME

Making large language models reliable data science programming copilots for biomedical research

Zifeng Wang, Benjamin Danek, Ziwei Yang, Zheng Chen, Jimeng Sun

Read Paper
Preprint

BIODSA-1K: Benchmarking Data Science Agents for Biomedical Research

Zifeng Wang, Benjamin Danek, Jimeng Sun

Read Paper
Nature Protocols

Tutorial: guidance on the use of large language models for medical research

Qiao Jin, Nicholas Wan, Robert Leaman, Shubo Tian, Zhizheng Wang, Yifan Yang, Zifeng Wang, Guangzhi Xiong, Po-Ting Lai, Qingqing Zhu, Benjamin Hou, Maame Sarfo-Gyamfi, Gongbo Zhang, Aidan Gilson, Balu Bhasuran, Zhe He, Aidong Zhang, Jimeng Sun, Chunhua Weng, Ronald M. Summers, Qingyu Chen, Yifan Peng, Zhiyong Lu

Read Paper
Scientific Data

MLOmics: Cancer Multi-Omics Database for Machine Learning

Ziwei Yang, Rikuto Kotoge, Xihao Piao, Zheng Chen, Lingwei Zhu, Peng Gao, Yasuko Matsubara, Yasushi Sakurai, Jimeng Sun

Read Paper
ICLR'24

BioBridge: Bridging Biomedical Foundation Models via Knowledge Graph

Zifeng Wang, Zichen Wang, Balasubramaniam Srinivasan, Vassilis N. Ioannidis, Huzefa Rangwala, Rishita Anubhai

Read Paper
EMNLP'22

MedCLIP: Contrastive Learning from Unpaired Medical Images and Text

Zifeng Wang, Jimeng Sun

Read Paper
Technical brochure

TrialMind for Biometric Teams

Keiji AI

Read Brochure (PDF)
Presentation

TrialMind for Biometric Teams — slide deck

Keiji AI

View Slides (PDF)
Technical brochure

TrialMind for Real-World Data and Evidence

Keiji AI

Read Brochure (PDF)
Presentation

TrialMind for Real-World Data and Evidence — slide deck

Keiji AI

View Slides (PDF)
Technical brochure

How LLM Training Actually Works: a practical guide to open-weight model training and evaluation

Jimeng Sun

Read Brochure (PDF)
Presentation

How LLM Training Actually Works — slide deck

Jimeng Sun

View Slides (PDF)