Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Cao Xiao

dblp:170/1833 · also Cao (Danica) Xiao · DBLP profile ↗
← Back
93ranked-venue papers
8as first author
53since 2021 · last 2026
0000-0002-3869-6942ORCID · corroborated

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 64 · 2 first-author · 39 since 2021Databases, data management, data science and information retrieval · 32 · 2 first-author · 17 since 2021Graphics, computer vision, multimedia, augmented reality and games · 26 · 1 first-author · 11 since 2021Applied, interdisciplinary, general and emerging computing · 23 · 3 first-author · 12 since 2021Human-computer interaction and ubiquitous computing · 2 · 2 first-authorSystems, architecture and hardware · 1 · 1 since 2021Software engineering, systems software and programming languages · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
47 papers
Language models and text generation · 16% Trustworthy machine learning · 12% Graph learning · 10%
Interdisciplinary, comprehensive, and emerging computing
48 papers
Medical and health informatics · 73% Bioinformatics and computational biology · 24% Computational science and engineering · 2%
Databases, data mining, and information retrieval
17 papers
Data mining · 54% Knowledge graphs · 27% Recommender systems · 8%
Network and information security
3 papers
Security and privacy of machine learning · 60% Privacy and data protection · 40%

Topics — the 30 heaviest of 154, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Medical and health informatics
clinical prediction
2.752025
Reasoning-Enhanced Healthcare Predictions with Knowledge Graph Community Retrieval · ICLR 2025
GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge Graphs · ICLR 2024
SafeDrug: Dual Molecular Graph Encoders for Recommending Effective and Safe Drug Combinations · IJCAI 2021
Bioinformatics and computational biology
drug discovery
2.252021
MolTrans: Molecular Interaction Transformer for drug-target interaction prediction · Bioinform. 2021
Probabilistic and Dynamic Molecule-Disease Interaction Modeling for Drug Discovery · KDD 2021
MIMOSA: Multi-constraint Molecule Sampling for Molecule Optimization · AAAI 2021
Medical and health informatics
electronic health records
1.942024
Recent Advances in Predictive Modeling with Electronic Health Records · IJCAI 2024
COMPOSE: Cross-Modal Pseudo-Siamese Network for Patient Trial Matching · KDD 2020
Longitudinal Adversarial Attack on Electronic Health Records Data · WWW 2019
Computer vision › Segmentation and scene understanding
medical image segmentation
1.722025
FlanS: A Foundation Model for Free-Form Language-based Segmentation in Medical Images · KDD (2) 2025
Enhancing SAM with Efficient Prompting and Preference Optimization for Semi-supervised Medical Image Segmentation · CVPR 2025
Data mining › multidimensional data analysis › multiway data analysis › tensor analysis
tensor factorization
1.632022
ATD: Augmenting CP Tensor Decomposition by Self Supervision · NeurIPS 2022
MTC: Multiresolution Tensor Completion from Partial and Coarse Observations · KDD 2021
STELAR: Spatio-temporal Tensor Factorization with Latent Epidemiological Regularization · AAAI 2021
Bioinformatics and computational biology › drug discovery
molecular optimization
1.532022
Differentiable Scaffolding Tree for Molecule Optimization · ICLR 2022
MIMOSA: Multi-constraint Molecule Sampling for Molecule Optimization · AAAI 2021
CORE: Automatic Molecule Optimization Using Copy & Refine Strategy · AAAI 2020
Medical and health informatics › clinical decision support
medication recommendation
1.542021
SafeDrug: Dual Molecular Graph Encoders for Recommending Effective and Safe Drug Combinations · IJCAI 2021
Change Matters: Medication Change Prediction with Recurrent Residual Networks · IJCAI 2021
GAMENet: Graph Augmented MEmory Networks for Recommending Medication Combination · AAAI 2019
Medical and health informatics › clinical prediction
health risk prediction
1.432021
MedPath: Augmenting Health Risk Prediction via Medical Knowledge Paths · WWW 2021
UNITE: Uncertainty-based Health Risk Prediction Leveraging Multi-sourced Data · WWW 2021
StageNet: Stage-Aware Neural Networks for Health Risk Prediction · WWW 2020
Machine learning › Trustworthy machine learning
uncertainty estimation
1.432024
Fast Online Value-Maximizing Prediction Sets with Conformal Cost Control · ICML 2023
Probabilistic and Dynamic Molecule-Disease Interaction Modeling for Drug Discovery · KDD 2021
Certifiably Byzantine-Robust Federated Conformal Prediction · ICML 2024
Machine learning › Generative modeling
molecular generation
1.332022
MOLER: Incorporate Molecule-Level Reward to Enhance Deep Generative Model for Molecule Optimization · IEEE Trans. Knowl. Data Eng. 2022
Differentiable Scaffolding Tree for Molecule Optimization · ICLR 2022
Probabilistic and Dynamic Molecule-Disease Interaction Modeling for Drug Discovery · KDD 2021
Privacy and data protection
privacy-preserving machine learning
1.322024
Certifiably Byzantine-Robust Federated Conformal Prediction · ICML 2024
AID: Active Distillation Machine to Leverage Pre-Trained Black-Box Models in Private Data Settings · WWW 2021
Machine learning › Graph learning
graph neural network
1.142024
GAMENet: Graph Augmented MEmory Networks for Recommending Medication Combination · AAAI 2019
FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling · ICLR (Poster) 2018
GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge Graphs · ICLR 2024
Bioinformatics and computational biology › drug discovery
drug-target interaction prediction
1.022021
MolTrans: Molecular Interaction Transformer for drug-target interaction prediction · Bioinform. 2021
DeepPurpose: a deep learning library for drug-target interaction prediction · Bioinform. 2021
Data mining › multidimensional data analysis › multiway data analysis › tensor analysis
tensor completion
1.022021
MTC: Multiresolution Tensor Completion from Partial and Coarse Observations · KDD 2021
Multi-version Tensor Completion for Time-delayed Spatio-temporal Data · IJCAI 2021
Medical and health informatics › drug safety
drug-drug interaction prediction
0.922021
SumGNN: multi-typed drug interaction prediction via efficient knowledge graph summarization · Bioinform. 2021
CASTER: Predicting Drug Interactions with Chemical Substructure Representation · AAAI 2020
Machine learning › Trustworthy machine learning › uncertainty estimation
conformal prediction
0.922024
Fast Online Value-Maximizing Prediction Sets with Conformal Cost Control · ICML 2023
Certifiably Byzantine-Robust Federated Conformal Prediction · ICML 2024
Machine learning › Trustworthy machine learning
interpretability
0.922021
AID: Active Distillation Machine to Leverage Pre-Trained Black-Box Models in Private Data Settings · WWW 2021
MINA: Multilevel Knowledge-Guided Attention for Modeling Electrocardiography Signals · IJCAI 2019
Machine learning › Efficient and distributed learning › model compression
knowledge distillation
0.922021
AID: Active Distillation Machine to Leverage Pre-Trained Black-Box Models in Private Data Settings · WWW 2021
RDPD: Rich Data Helps Poor Data via Imitation · IJCAI 2019
Machine learning › Deep learning architectures and training › sequence modeling
continuous-time sequence modeling
0.912025
Deep Continuous-Time State-Space Models for Marked Event Sequences · NeurIPS 2025
Machine learning › Representation and self-supervised learning › pre-training › unsupervised pre-training
contrastive pre-training
0.912025
Bi-level Contrastive Learning for Knowledge-Enhanced Molecule Representations · AAAI 2025
Knowledge, reasoning and agents › Knowledge representation and reasoning
knowledge graph
0.912025
A Survey on Unifying Large Language Models and Knowledge Graphs for Biomedicine and Healthcare · KDD (2) 2025
Knowledge, reasoning and agents › Knowledge representation and reasoning › semantic representation › grounding
knowledge grounding
0.912025
A Survey on Unifying Large Language Models and Knowledge Graphs for Biomedicine and Healthcare · KDD (2) 2025
Computer vision › Vision and language › visual grounding
language-guided segmentation
0.912025
FlanS: A Foundation Model for Free-Form Language-based Segmentation in Medical Images · KDD (2) 2025
Natural language and speech › Language models and text generation
large language model
0.912025
A Survey on Unifying Large Language Models and Knowledge Graphs for Biomedicine and Healthcare · KDD (2) 2025
Computer vision › Vision and language › vision-language model › domain-specific vision-language model
medical vision-language model
0.912025
Focus on What Matters: Enhancing Medical Vision-Language Models with Automatic Attention Alignment Tuning · ACL (1) 2025
Machine learning › Graph learning › graph representation learning
molecular graph representation
0.912025
Bi-level Contrastive Learning for Knowledge-Enhanced Molecule Representations · AAAI 2025
Computer vision › Vision and language › vision-language model
prompt learning
0.912025
Enhancing SAM with Efficient Prompting and Preference Optimization for Semi-supervised Medical Image Segmentation · CVPR 2025
Computer vision › Segmentation and scene understanding › medical image segmentation
semi-supervised segmentation
0.912025
Enhancing SAM with Efficient Prompting and Preference Optimization for Semi-supervised Medical Image Segmentation · CVPR 2025
Machine learning › Deep learning architectures and training
state space model
0.912025
Deep Continuous-Time State-Space Models for Marked Event Sequences · NeurIPS 2025
Machine learning › Probabilistic and Bayesian machine learning › stochastic processes › point process
temporal point process
0.912025
Deep Continuous-Time State-Space Models for Marked Event Sequences · NeurIPS 2025

Methods — techniques the papers use, named apart from their topics

graph neural network · 7.8large language model · 4.9retrieval-augmented generation · 4.4knowledge distillation · 2.9conformal prediction · 2.8attention mechanism · 2.7fine-tuning · 2.4deep learning · 2.1adversarial training · 1.7recurrent neural network · 1.5alternating least squares · 1.1transformer · 0.9knowledge graph · 0.9hierarchical embedding · 0.9graph community detection · 0.9contrastive learning · 0.9contrastive language-image pretraining · 0.9spectral regularization · 0.9
YearPublicationVenuePosition
2026 Fault-tolerance Mapping of Spiking Neural Networks to RRAM-based Neuromorphic Hardware
abstract
Spiking neural networks (SNNs) have been widely used in artificial intelligence applications. Resistive random-access memory (RRAM) based neuromorphic hardware can be used for low-power and high-speed inference of SNNs. However, RRAM devices suffer from stuck-at-fault (SAF) defects due to the immature fabrication process. SAF defects can lead to incorrect weights of SNNs and thus severely degrade the inference accuracy. In this paper, we propose a fault-tolerance synaptic-to-RRAM mapping scheme to deploy SNNs while protecting the inference accuracy. We first explore how different weights affect the model accuracy in SNNs and find that the frequency of spikes plays a vital role. Motivated by this, we develop an SNN-oriented metric to evaluate the importance of weights. Then we propose a defect-aware mapping scheme based on a simulated-annealing framework to efficiently map synapses to RRAM to mitigate the impact of SAFs. Evaluation shows that the proposed strategy can improve the accuracy by 18.74% on average compared to existing mapping strategy.
Yuqing Xiong, Cao Xiao, Mengying Zhao
DATE2
2025 Bi-level Contrastive Learning for Knowledge-Enhanced Molecule Representations
abstract
Molecular representation learning is vital for various downstream applications, including the analysis and prediction of molecular properties and side effects. While Graph Neural Networks (GNNs) have been a popular framework for modeling molecular data, they often struggle to capture the full complexity of molecular representations. In this paper, we introduce a novel method called Gode, which accounts for the dual-level structure inherent in molecules. Molecules possess an intrinsic graph structure and simultaneously function as nodes within a broader molecular knowledge graph. Gode integrates individual molecular graph representations with multi-domain biochemical data from knowledge graphs. By pre-training two GNNs on different graph structures and employing contrastive learning, Gode effectively fuses molecular structures with their corresponding knowledge graph substructures. This fusion yields a more robust and informative representation, enhancing molecular property predictions by leveraging both chemical and biological information. When fine-tuned across 11 chemical property tasks, our model significantly outperforms existing benchmarks, achieving an average ROC-AUC improvement of 12.7% for classification tasks and an average RMSE/MAE improvement of 34.4% for regression tasks. Notably, Gode surpasses the current leading model in property prediction, with advancements of 2.2% in classification and 7.2% in regression tasks.
Pengcheng Jiang, Cao Xiao, Tianfan Fu, Parminder Bhatia, Taha A. Kass-Hout, Jimeng Sun 0001, Jiawei Han 0001
AAAI2
2025 Focus on What Matters: Enhancing Medical Vision-Language Models with Automatic Attention Alignment Tuning
abstract
Aofei Chang, Le Huang, Alex James Boyd, Parminder Bhatia, Taha Kass-Hout, Cao Xiao, Fenglong Ma. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025.
Aofei Chang, Alex Boyd, Parminder Bhatia, Taha A. Kass-Hout, Cao Xiao, Fenglong Ma
ACL (1)6
2025 Enhancing SAM with Efficient Prompting and Preference Optimization for Semi-supervised Medical Image Segmentation
abstract
Foundational models such as the Segment Anything Model (SAM) are gaining traction in medical imaging segmentation, supporting multiple downstream tasks. However, such models are supervised in nature, still relying on large annotated datasets or prompts supplied by experts. Conventional techniques such as active learning to alleviate such limitations are limited in scope and still necessitate continuous human involvement and complex domain knowledge for label refinement or establishing reward ground truth. To address these challenges, we propose an enhanced Segment Anything Model (SAM) framework that utilizes annotation-efficient prompts generated in a fully unsupervised fashion, while still capturing essential semantic, location, and shape information through contrastive language-image pretraining and visual question answering. We adopt the direct preference optimization technique to design an optimal policy that enables the model to generate high-fidelity segmentations with simple ratings or rankings provided by a virtual annotator simulating the human annotation process. State-of-the-art performance of our framework in tasks such as lung segmentation, breast tumor segmentation, and organ segmentation across various modalities, including X-ray, ultrasound, and abdominal CT, justifies its effectiveness in low-annotation data scenarios.
Aishik Konwer, Zhijian Yang, Erhan Bas, Cao Xiao, Prateek Prasanna, Parminder Bhatia, Taha A. Kass-Hout
CVPR4
2025 Reasoning-Enhanced Healthcare Predictions with Knowledge Graph Community Retrieval
abstract
Large language models (LLMs) have demonstrated significant potential in clinical decision support. Yet LLMs still suffer from hallucinations and lack fine-grained contextual medical knowledge, limiting their high-stake healthcare applications such as clinical diagnosis. Traditional retrieval-augmented generation (RAG) methods attempt to address these limitations but frequently retrieve sparse or irrelevant information, undermining prediction accuracy. We introduce KARE, a novel framework that integrates knowledge graph (KG) community-level retrieval with LLM reasoning to enhance healthcare predictions. KARE constructs a comprehensive multi-source KG by integrating biomedical databases, clinical literature, and LLM-generated insights, and organizes it using hierarchical graph community detection and summarization for precise and contextually relevant information retrieval. Our key innovations include: (1) a dense medical knowledge structuring approach enabling accurate retrieval of relevant information; (2) a dynamic knowledge retrieval mechanism that enriches patient contexts with focused, multi-faceted medical insights; and (3) a reasoning-enhanced prediction framework that leverages these enriched contexts to produce both accurate and interpretable clinical predictions. Extensive experiments demonstrate that KARE outperforms leading models by up to 10.8-15.0\% on MIMIC-III and 12.6-12.7\% on MIMIC-IV for mortality and readmission predictions. In addition to its impressive prediction accuracy, our framework leverages the reasoning capabilities of LLMs, enhancing the trustworthiness of clinical predictions.
Pengcheng Jiang, Cao Xiao, Minhao Jiang, Parminder Bhatia, Taha A. Kass-Hout, Jimeng Sun 0001, Jiawei Han 0001
ICLR2
2025 A Survey on Unifying Large Language Models and Knowledge Graphs for Biomedicine and Healthcare
abstract
In recent years, the landscape of digital biomedicine and healthcare has been reshaped due to the disruptive breakthroughs in AIfacilitated by tremendous data and high-performance computers, large language models (LLMs) have transformed information technology from accessing data to performing analytical tasks.While demonstrating unprecedented capabilities, LLMs have been found unreliable in tasks requiring factual knowledge and rigorous reasoning.Biomedicine and healthcare, as an important vertical domain rapidly benefitting from progress in AI, necessitates strict requirements on the accuracy, controllability, and interpretability of analytical models, posing critical challenges for LLMs.Despite recent studies addressing the hallucination problem of LLMs, research on empowering LLMs with the ability to plan, reason, and ground with explicit knowledge has also started to prosper, especially in the biomedicine and healthcare domain.On the other hand, biomedical data are enormous and notoriously complex, coming from various sources (e.g., biomedical knowledge bases, online literature, and hospitals) and bearing various modalities (e.g., tables, texts, images and time-series).Healthcare professionals have spent decades collecting, cleaning, and curating various types of data.The processes are extremely costly, producing various datasets with different data schemas, coding systems, and quality standards, many privately
Ran Xu 0002, Patrick Jiang, Linhao Luo, Cao Xiao, Adam R. Cross, Shirui Pan, Jimeng Sun 0001, Carl Yang 0001
KDD (2)4
2025 FlanS: A Foundation Model for Free-Form Language-based Segmentation in Medical Images
abstract
KDD ’25, August 3–7, 2025, Toronto, ON, Canada
Longchao Da, Rui Wang 0184, Xiaojian Xu 0002, Parminder Bhatia, Taha A. Kass-Hout, Hua Wei 0001, Cao Xiao
KDD (2)7
2025 Dynamic Uncertainty Ranking: Enhancing Retrieval-Augmented In-Context Learning for Long-Tail Knowledge in LLMs
abstract
Shuyang Yu, Runxue Bao, Parminder Bhatia, Taha Kass-Hout, Jiayu Zhou, Cao Xiao. Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2025.
Shuyang Yu, Runxue Bao, Parminder Bhatia, Taha A. Kass-Hout, Cao Xiao
NAACL (Long Papers)6
2025 Deep Continuous-Time State-Space Models for Marked Event Sequences
abstract
Marked temporal point processes (MTPPs) model sequences of events occurring at irregular time intervals, with wide-ranging applications in fields such as healthcare, finance and social networks. We propose the _state-space point process_ (S2P2) model, a novel and performant model that leverages techniques derived for modern deep state-space models (SSMs) to overcome limitations of existing MTPP models, while simultaneously imbuing strong inductive biases for continuous-time event sequences that other discrete sequence models (i.e., RNNs, transformers) do not capture. Inspired by the classical linear Hawkes processes, we propose an architecture that interleaves stochastic jump differential equations with nonlinearities to create a highly expressive intensity-based MTPP model, without the need for restrictive parametric assumptions for the intensity. Our approach enables efficient training and inference with a parallel scan, bringing linear complexity and sublinear scaling while retaining expressivity to MTPPs. Empirically, S2P2 achieves state-of-the-art predictive likelihoods across eight real-world datasets, delivering an average improvement of 33% over the best existing approaches.
Yuxin Chang, Alex Boyd, Cao Xiao, Taha A. Kass-Hout, Parminder Bhatia, Padhraic Smyth, Andrew Warrington
NeurIPS3
2025 Any Large Language Model Can Be a Reliable Judge: Debiasing with a Reasoning-based Bias Detector
abstract
LLM-as-a-Judge has emerged as a promising tool for automatically evaluating generated outputs, but its reliability is often undermined by potential biases in judgment. Existing efforts to mitigate these biases face key limitations: in-context learning-based methods fail to address rooted biases due to the evaluator’s limited capacity for self-reflection, whereas fine-tuning is not applicable to all evaluator types, especially closed-source models. To address this challenge, we introduce the **R**easoning-based **B**ias **D**etector (RBD), which is a plug-in module that identifies biased evaluations and generates structured reasoning to guide evaluator self-correction. Rather than modifying the evaluator itself, RBD operates externally and engages in an iterative process of bias detection and feedback-driven revision. To support its development, we design a complete pipeline consisting of biased dataset construction, supervision collection, distilled reasoning-based fine-tuning of RBD, and integration with LLM evaluators. We fine-tune four sizes of RBD models, ranging from 1.5B to 14B, and observe consistent performance improvements across all scales. Experimental results on 4 bias types—verbosity, position, bandwagon, and sentiment—evaluated using 8 LLM evaluators demonstrate RBD’s strong effectiveness. For example, the RBD-8B model improves evaluation accuracy by an average of 18.5% and consistency by 10.9%, and surpasses prompting-based baselines and fine-tuned judges by 12.8% and 17.2%, respectively. These results highlight RBD’s effectiveness and scalability. Additional experiments further demonstrate its strong generalization across biases and domains, as well as its efficiency.
Runxue Bao, Cao Xiao, Parminder Bhatia, Shangqian Gao, Taha A. Kass-Hout
NeurIPS3
2024 ConSequence: Synthesizing Logically Constrained Sequences for Electronic Health Record Generation
abstract
Generative models can produce synthetic patient records for analytical tasks when real data is unavailable or limited. However, current methods struggle with adhering to domain-specific knowledge and removing invalid data. We present ConSequence, an effective approach to integrating domain knowledge into sequential generative neural network outputs. Our rule-based formulation includes temporal aggregation and antecedent evaluation modules, ensured by an efficient matrix multiplication formulation, to satisfy hard and soft logical constraints across time steps. Existing constraint methods often fail to guarantee constraint satisfaction, lack the ability to handle temporal constraints, and hinder the learning and computational efficiency of the model. In contrast, our approach efficiently handles all types of constraints with guaranteed logical coherence. We demonstrate ConSequence's effectiveness in generating electronic health records, outperforming competitors in achieving complete temporal and spatial constraint satisfaction without compromising runtime performance or generative quality. Specifically, ConSequence successfully prevents all rule violations while improving the model quality in reducing its test perplexity by 5% and incurring less than a 13% slowdown in generation speed compared to an unconstrained model.
Brandon Theodorou, Shrusti Jain, Cao Xiao, Jimeng Sun 0001
AAAI3
2024 Unity in Diversity: Collaborative Pre-training Across Multimodal Medical Sources
abstract
re-training (MEDCSP), a new pre-training strategy designed to bridge the gap between multimodal medical sources. MEDCSP employs modality-level aggregation to unify patient data within individual sources. Additionally, leveraging temporal information and diagnosis history, MEDCSP effectively captures explicit and implicit correlations between patients across different sources. To evaluate the proposed strategy, we conduct comprehensive experiments, where the experiments are based on 6 modalities from 2 real-world medical data sources, and MEDCSP is evaluated on 4 tasks against 19 baselines, marking an initial yet essential step towards cross-source modeling in the medical domain.
Xiaochen Wang 0002, Junyu Luo 0001, Jiaqi Wang 0002, Yuan Zhong 0002, Xiaokun Zhang 0001, Yaqing Wang 0001, Parminder Bhatia, Cao Xiao, Fenglong Ma
ACL (1)8
2024 Unlocking Memorization in Large Language Models with Dynamic Soft Prompting
abstract
Zhepeng Wang, Runxue Bao, Yawen Wu, Jackson Taylor, Cao Xiao, Feng Zheng, Weiwen Jiang, Shangqian Gao, Yanfu Zhang. Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing. 2024.
Zhepeng Wang 0001, Runxue Bao, Yawen Wu, Jackson Taylor, Cao Xiao, Feng Zheng 0001, Weiwen Jiang, Shangqian Gao, Yanfu Zhang
EMNLP5
2024 GraphCare: Enhancing Healthcare Predictions with Personalized Knowledge Graphs
abstract
Clinical predictive models often rely on patients’ electronic health records (EHR), but integrating medical knowledge to enhance predictions and decision-making is challenging. This is because personalized predictions require personalized knowledge graphs (KGs), which are difficult to generate from patient EHR data. To address this, we propose GraphCare, an open-world framework that uses external KGs to improve EHR-based predictions. Our method extracts knowledge from large language models (LLMs) and external biomedical KGs to build patient-specific KGs, which are then used to train our proposed Bi-attention AugmenTed (BAT) graph neural network (GNN) for healthcare predictions. On two public datasets, MIMIC-III and MIMIC-IV, GraphCare surpasses baselines in four vital healthcare prediction tasks: mortality, readmission, length of stay (LOS), and drug recommendation. On MIMIC-III, it boosts AUROC by 17.6% and 6.6% for mortality and readmission, and F1-score by 7.9% and 10.8% for LOS and drug recommendation, respectively. Notably, GraphCare demonstrates a substantial edge in scenarios with limited data availability. Our findings highlight the potential of using external KGs in healthcare prediction tasks and demonstrate the promise of GraphCare in generating personalized KGs for promoting personalized medicine.
Pengcheng Jiang, Cao Xiao, Adam R. Cross, Jimeng Sun 0001
ICLR2
2024 Certifiably Byzantine-Robust Federated Conformal Prediction
abstract
Conformal prediction has shown impressive capacity in constructing statistically rigorous prediction sets for machine learning models with exchangeable data samples. The siloed datasets, coupled with the escalating privacy concerns related to local data sharing, have inspired recent innovations extending conformal prediction into federated environments with distributed data samples. However, this framework for distributed uncertainty quantification is susceptible to Byzantine failures. A minor subset of malicious clients can significantly compromise the practicality of coverage guarantees. To address this vulnerability, we introduce a novel framework Rob-FCP, which executes robust federated conformal prediction, effectively countering malicious clients capable of reporting arbitrary statistics with the conformal calibration process. We theoretically provide the conformal coverage bound of Rob-FCP in the Byzantine setting and show that the coverage of Rob-FCP is asymptotically close to the desired coverage level. We also propose a malicious client number estimator to tackle a more challenging setting where the number of malicious clients is unknown to the defender and theoretically shows its effectiveness. We empirically demonstrate the robustness of Rob-FCP against diverse proportions of malicious clients under a variety of Byzantine attacks on five standard benchmark and real-world healthcare datasets.
Mintong Kang, Zhen Lin 0001, Jimeng Sun 0001, Cao Xiao, Bo Li 0026
ICML4
2024 Recent Advances in Predictive Modeling with Electronic Health Records
Jiaqi Wang 0002, Junyu Luo 0001, Muchao Ye, Xiaochen Wang 0002, Yuan Zhong 0002, Aofei Chang, Guanjie Huang, Ziyi Yin 0003, Cao Xiao, Jimeng Sun 0001, Fenglong Ma
IJCAI9
2024 MediTab: Scaling Medical Tabular Data Predictors via Data Consolidation, Enrichment, and Refinement
Zifeng Wang 0008, Chufan Gao, Cao Xiao, Jimeng Sun 0001
IJCAI3
2024 Synthesizing Multimodal Electronic Health Records via Predictive Diffusion Models
abstract
Synthesizing electronic health records (EHR) data has become a preferred strategy to address data scarcity, improve data quality, and model fairness in healthcare. However, existing approaches for EHR data generation predominantly rely on state-of-the-art generative techniques like generative adversarial networks, variational autoencoders, and language models. These methods typically replicate input visits, resulting in inadequate modeling of temporal dependencies between visits and overlooking the generation of time information, a crucial element in EHR data. Moreover, their ability to learn visit representations is limited due to simple linear mapping functions, thus compromising generation quality. To address these limitations, we propose a novel EHR data generation model called EHRPD. It is a diffusion-based model designed to predict the next visit based on the current one while also incorporating time interval estimation. To enhance generation quality and diversity, we introduce a novel time-aware visit embedding module and a pioneering predictive denoising diffusion probabilistic model (P-DDPM). Additionally, we devise a predictive U-Net (PU-Net) to optimize P-DDPM. We conduct experiments on two public datasets and evaluate EHRPD from fidelity, privacy, and utility perspectives. The experimental results demonstrate the efficacy and utility of the proposed EHRPD in addressing the aforementioned limitations and advancing EHR data generation.
Yuan Zhong 0002, Xiaochen Wang 0002, Jiaqi Wang 0002, Xiaokun Zhang 0001, Yaqing Wang 0001, Mengdi Huai, Cao Xiao, Fenglong Ma
KDD7
2024 PILOT: Legal Case Outcome Prediction with Case Law
abstract
Lang Cao, Zifeng Wang, Cao Xiao, Jimeng Sun. Proceedings of the 2024 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2024.
Lang Cao, Zifeng Wang 0008, Cao Xiao, Jimeng Sun 0001
NAACL-HLT3
2024 TriSum: Learning Summarization Ability from Large Language Models with Structured Rationale
abstract
Pengcheng Jiang, Cao Xiao, Zifeng Wang, Parminder Bhatia, Jimeng Sun, Jiawei Han. Proceedings of the 2024 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2024.
Pengcheng Jiang, Cao Xiao, Zifeng Wang 0008, Parminder Bhatia, Jimeng Sun 0001, Jiawei Han 0001
NAACL-HLT2
2024 KG-FIT: Knowledge Graph Fine-Tuning Upon Open-World Knowledge
abstract
Knowledge Graph Embedding (KGE) techniques are crucial in learning compact representations of entities and relations within a knowledge graph, facilitating efficient reasoning and knowledge discovery. While existing methods typically focus either on training KGE models solely based on graph structure or fine-tuning pre-trained language models with classification data in KG, KG-FIT leverages LLM-guided refinement to construct a semantically coherent hierarchical structure of entity clusters. By incorporating this hierarchical knowledge along with textual information during the fine-tuning process, KG-FIT effectively captures both global semantics from the LLM and local semantics from the KG. Extensive experiments on the benchmark datasets FB15K-237, YAGO3-10, and PrimeKG demonstrate the superiority of KG-FIT over state-of-the-art pre-trained language model-based methods, achieving improvements of 14.4\%, 13.5\%, and 11.9\% in the Hits@10 metric for the link prediction task, respectively. Furthermore, KG-FIT yields substantial performance gains of 12.6\%, 6.7\%, and 17.7\% compared to the structure-based base models upon which it is built. These results highlight the effectiveness of KG-FIT in incorporating open-world knowledge from LLMs to significantly enhance the expressiveness and informativeness of KG embeddings.
Pengcheng Jiang, Lang Cao, Cao Xiao, Parminder Bhatia, Jimeng Sun 0001, Jiawei Han 0001
NeurIPS3
2023 ClinicalRisk: A New Therapy-related Clinical Trial Dataset for Predicting Trial Status and Failure Reasons
abstract
Clinical trials aim to study new tests and evaluate their effects on human health outcomes, which has a huge market size. However, carrying out clinical trials is expensive and time-consuming and often ends in no results. It will revolutionize clinical practice if we can develop an effective model to automatically estimate the status of a clinical trial and find out possible failure reasons. However, it is challenging to develop such a model because of the lack of a benchmark dataset. To address these challenges, in this paper, we first build a new dataset by extracting the publicly available clinical trial reports from ClinicalTrials.gov. The associated status of each report is treated as the status label. To analyze the failure reasons, domain experts help us manually annotate each failed report based on the description associated with it. More importantly, we examine several state-of-the-art text classification baselines on this task and find out that the unique format of the clinical trial protocols plays an essential role in affecting prediction accuracy, demonstrating the need for specially designed clinical trial classification models.
Junyu Luo 0001, Zhi Qiao 0007, Lucas Glass, Cao Xiao, Fenglong Ma
CIKM4
2023 AutoTrial: Prompting Language Models for Clinical Trial Design
abstract
Clinical trials are critical for drug development.Constructing the appropriate eligibility criteria (i.e., the inclusion/exclusion criteria for patient recruitment) is essential for the trial's success.Proper design of clinical trial protocols should consider similar precedent trials and their eligibility criteria to ensure sufficient patient coverage.In this paper, we present a method named AutoTrial to aid the design of clinical eligibility criteria using language models.It allows (1) controllable generation under instructions via a hybrid of discrete and neural prompting, (2) scalable knowledge incorporation via in-context learning, and (3) explicit reasoning chains to provide rationales for understanding the outputs.Experiments on over 70K clinical trials verify that AutoTrial generates high-quality criteria texts that are fluent and coherent and with high accuracy in capturing the relevant clinical concepts to the target trial.It is noteworthy that our method, with a much smaller parameter size, gains around 60% winning rate against the GPT-3.5 baselines via human evaluations.
Zifeng Wang 0008, Cao Xiao, Jimeng Sun 0001
EMNLP2
2023 Fast Online Value-Maximizing Prediction Sets with Conformal Cost Control
abstract
Many real-world multi-label prediction problems involve set-valued predictions that must satisfy specific requirements dictated by downstream usage. We focus on a typical scenario where such requirements, separately encoding *value* and *cost*, compete with each other. For instance, a hospital might expect a smart diagnosis system to capture as many severe, often co-morbid, diseases as possible (the value), while maintaining strict control over incorrect predictions (the cost). We present a general pipeline, dubbed as FavMac, to maximize the value while controlling the cost in such scenarios. FavMac can be combined with almost any multi-label classifier, affording distribution-free theoretical guarantees on cost control. Moreover, unlike prior works, FavMac can handle real-world large-scale applications via a carefully designed online update mechanism, which is of independent interest. Our methodological and theoretical contributions are supported by experiments on several healthcare tasks and synthetic datasets - FavMac furnishes higher value compared with several variants and baselines while maintaining strict cost control.
Zhen Lin 0001, Shubhendu Trivedi, Cao Xiao, Jimeng Sun 0001
ICML3
2023 MedLink: De-Identified Patient Health Record Linkage
abstract
A comprehensive patient health history is essential for patient care and healthcare research. However, due to the distributed nature of healthcare services, patient health records are often scattered across multiple systems. Existing record linkage approaches primarily rely on patient identifiers, which have inherent limitations such as privacy invasion and identifier discrepancies. To tackle this problem, we propose linking de-identified patient health records by matching health patterns without strictly relying on sensitive patient identifiers. Our model MedLink solves two challenges faced with the patient linkage task: (1) the challenge of identifying the same patients based on data collected in different timelines as disease progression makes the record matching difficult, and (2) the challenge of identifying distinct health patterns as common medical codes dominate health records and overshadow the more informative low-prevalence codes. To address these challenges, MedLink utilizes bi-directional health prediction to predict future codes forwardly and past codes backwardly, thus accounting for the health progression. MedLink also has a prevalence-aware retrieval design to focus more on the low-prevalence but informative codes during learning. MedLink can be trained end-to-end and is lightweight for efficient inference on large patient databases. We evaluate MedLink against leading baselines on real-world patient datasets, including the critical care dataset MIMIC-III and a large health claims dataset. Results show that MedLink outperforms the best baseline by 4% in top-1 accuracy with only 8% memory cost. Additionally, when combined with existing identifier-based linkage approaches, MedLink can improve their performance by up to 15%.
Zhenbang Wu, Cao Xiao, Jimeng Sun 0001
KDD2
2022 SCRIB: Set-Classifier with Class-Specific Risk Bounds for Blackbox Models
abstract
Despite deep learning (DL) success in classification problems, DL classifiers do not provide a sound mechanism to decide when to refrain from predicting. Recent works tried to control the overall prediction risk with classification with rejection options. However, existing works overlook the different significance of different classes. We introduce Set-classifier with class-specific RIsk Bounds (SCRIB) to tackle this problem, assigning multiple labels to each example. Given the output of a black-box model on the validation set, SCRIB constructs a set-classifier that controls the class-specific prediction risks. The key idea is to reject when the set classifier returns more than one label. We validated SCRIB on several medical applications, including sleep staging on electroencephalogram(EEG) data, X-ray COVID image classification, and atrial fibrillation detection based on electrocardiogram (ECG) data.SCRIB obtained desirable class-specific risks, which are 35%-88% closer to the target risks than baseline methods.
Zhen Lin 0001, Lucas Glass, M. Brandon Westover, Cao Xiao, Jimeng Sun 0001
AAAI4
2022 MedAttacker: Exploring Black-Box Adversarial Attacks on Risk Prediction Models in Healthcare
abstract
Researchers have conduct adversarial attacks against deep neural networks (DNNs) for health risk prediction in the white/gray-box setting to evaluate their robustness. However, since most real-world solutions are trained by private data and released as black-box services on the cloud, we should investigate their robustness in the black-box setting. Unfortunately, existing work ignores to consider the uniqueness of electronic health records (EHRs). To fill this gap, we propose the first black-box adversarial attack method against health risk prediction models named MedAttacker to investigate their vulnerability. It addresses the challenges brought by EHRs via two steps: hierarchical position selection which selects the attacked positions in a reinforcement learning (RL) framework and substitute selection which identifies substitutes with a score-based principle. Particularly, by considering the temporal context inside EHRs, MedAttacker initializes its RL position selection policy by using the contribution score of each visit and the saliency score of each code, which can be well integrated with the deterministic substitute selection process decided by the score changes. We evaluate MedAttacker by attacking three advanced risk prediction models in the black-box setting across multiple real-world datasets, and MedAttacker consistently achieves the highest average success rate and even outperforms a recent white-box EHR adversarial attack technique in certain cases.
Muchao Ye, Junyu Luo 0001, Guanjie Zheng, Cao Xiao, Houping Xiao, Ting Wang 0006, Fenglong Ma
BIBM4
2022 Benchmarking Automated Clinical Language Simplification: Dataset, Algorithm, and Evaluation
abstract
Patients with low health literacy usually have difficulty understanding medical jargon and the complex structure of professional medical language. Although some studies are proposed to automatically translate expert language into layperson-understandable language, only a few of them focus on both accuracy and readability aspects simultaneously in the clinical domain. Thus, simplification of the clinical language is still a challenging task, but unfortunately, it is not yet fully addressed in previous work. To benchmark this task, we construct a new dataset named MedLane to support the development and evaluation of automated clinical language simplification approaches. Besides, we propose a new model called DECLARE that follows the human annotation procedure and achieves state-of-the-art performance compared with eight strong baselines. To fairly evaluate the performance, we also propose three specific evaluation metrics. Experimental results demonstrate the utility of the annotated MedLane dataset and the effectiveness of the proposed model DECLARE.
Junyu Luo 0001, Junxian Lin, Chi Lin 0001, Cao Xiao, Xinning Gui, Fenglong Ma
COLING4
2022 Differentiable Scaffolding Tree for Molecule Optimization
Tianfan Fu, Wenhao Gao 0001, Cao Xiao, Jacob Yasonik, Connor W. Coley, Jimeng Sun 0001
ICLR3
2022 ATD: Augmenting CP Tensor Decomposition by Self Supervision
abstract
Tensor decompositions are powerful tools for dimensionality reduction and feature interpretation of multidimensional data such as signals. Existing tensor decomposition objectives (e.g., Frobenius norm) are designed for fitting raw data under statistical assumptions, which may not align with downstream classification tasks. In practice, raw input tensor can contain irrelevant information while data augmentation techniques may be used to smooth out class-irrelevant noise in samples. This paper addresses the above challenges by proposing augmented tensor decomposition (ATD), which effectively incorporates data augmentations and self-supervised learning (SSL) to boost downstream classification. To address the non-convexity of the new augmented objective, we develop an iterative method that enables the optimization to follow an alternating least squares (ALS) fashion. We evaluate our proposed ATD on multiple datasets. It can achieve 0.8%~2.5% accuracy gain over tensor-based baselines. Also, our ATD model shows comparable or better performance (e.g., up to 15% in accuracy) over self-supervised and autoencoder baselines while using less than 5% of learnable parameters of these baseline models.
Chaoqi Yang, Cheng Qian 0001, Cao Xiao, M. Brandon Westover, Edgar Solomonik, Jimeng Sun 0001
NeurIPS4
2022 AutoMap: Automatic Medical Code Mapping for Clinical Prediction Model Deployment
Zhenbang Wu, Cao Xiao, Lucas Glass, David M. Liebovitz, Jimeng Sun 0001
ECML/PKDD (2)2
2022 PopNet: Real-Time Population-Level Disease Prediction with Data Latency
abstract
Population-level disease prediction estimates the number of potential patients of particular diseases in some location at a future time based on (frequently updated) historical disease statistics. Existing approaches often assume the existing disease statistics are reliable and will not change. However, in practice, data collection is often time-consuming and has time delays, with both historical and current disease statistics being updated continuously. In this work, we propose a real-time population-level disease prediction model which captures data latency (PopNet) and incorporates the updated data for improved predictions. To achieve this goal, PopNet models real-time data and updated data using two separate systems, each capturing spatial and temporal effects using hybrid graph attention networks and recurrent neural networks. PopNet then fuses the two systems using both spatial and temporal latency-aware attentions in an end-to-end manner. We evaluate PopNet on real-world disease datasets and show that PopNet consistently outperforms all baseline disease prediction and general spatial-temporal prediction models, achieving up to 47% lower root mean squared error and 24% lower mean absolute error compared with the best baselines.
Cao Xiao, Lucas Glass, Jimeng Sun 0001
WWW2
2022 MOLER: Incorporate Molecule-Level Reward to Enhance Deep Generative Model for Molecule Optimization
abstract
The goal of molecular optimization is to generate molecules similar to a target molecule but with better chemical properties. Deep generative models have shown great success in molecule optimization. However, due to the iterative local generation process of deep generative models, the resulting molecules can significantly deviate from the input in molecular similarity and size, leading to poor chemical properties. The key issue here is that the existing deep generative models restrict their attention on substructure-level generation without considering the entire molecule as a whole. To address this challenge, we propose Molecule-Level Reward functions (MOLER) to encourage (1) the input and the generated molecule to be similar, and to ensure (2) the generated molecule has a similar size to the input. The proposed method can be combined with various deep generative models. Policy gradient technique is introduced to optimize reward-based objectives with small computational overhead. Empirical studies show that MOLER achieves up to 20.2% relative improvement in success rate over the best baseline method on several properties, including QED, DRD2 and LogP.
Tianfan Fu, Cao Xiao, Lucas Glass, Jimeng Sun 0001
IEEE Trans. Knowl. Data Eng.2
2022 CHEER: Rich Model Helps Poor Model via Knowledge Infusion
abstract
There is a growing interest in applying deep learning (DL) to healthcare, driven by the availability of data with multiple feature channels inrich-dataenvironments (e.g., intensive care units). However, in many other practical situations, we can only access data with much fewer feature channels in apoor-dataenvironments (e.g., at home), which often results in predictive models with poor performance. How can we boost the performance of models learned from suchpoor-dataenvironment by leveraging knowledge extracted from existing models trained usingrich datain a related environment? To address this question, we develop a knowledge infusion framework namedCHEERthat can succinctly summarize suchrich modelinto transferable representations, which can be incorporated into thepoor modelto improve its performance. The infused model is analyzed theoretically and evaluated empirically on several datasets. Our empirical results showed thatCHEERoutperformed baselines by 5.60 to 46.80 percent in terms of the macro-F1 score on multiple physiological datasets.
Cao Xiao, Trong Nghia Hoang, Shenda Hong, Tengfei Ma 0001, Jimeng Sun 0001
IEEE Trans. Knowl. Data Eng.1
2021 MIMOSA: Multi-constraint Molecule Sampling for Molecule Optimization
abstract
Molecule optimization is a fundamental task for accelerating drug discovery, with the goal of generating new valid molecules that maximize multiple drug properties while maintaining similarity to the input molecule. Existing generative models and reinforcement learning approaches made initial success, but still face difficulties in simultaneously optimizing multiple drug properties. To address such challenges, we propose the MultI-constraint MOlecule SAmpling (MIMOSA) approach, a sampling framework to use input molecule as an initial guess and sample molecules from the target distribution. MIMOSA first pretrains two property agnostic graph neural networks (GNNs) for molecule topology and substructure-type prediction, where a substructure can be either atom or single ring. For each iteration, MIMOSA uses the GNNs’ prediction and employs three basic substructure operations (add, replace, delete) to generate new molecules and associated weights. The weights can encode multiple constraints including similarity and drug property constraints, upon which we select promising molecules for next iteration. MIMOSA enables flexible encoding of multiple property- and similarity-constraints and can efficiently generate new molecules that satisfy various property constraints and achieved up to 49.1% relative improvement over the best baseline in terms of success rate.
Tianfan Fu, Cao Xiao, Xinhao Li 0001, Lucas Glass, Jimeng Sun 0001
AAAI2
2021 STELAR: Spatio-temporal Tensor Factorization with Latent Epidemiological Regularization
abstract
Accurate prediction of the transmission of epidemic diseases such as COVID-19 is crucial for implementing effective mitigation measures. In this work, we develop a tensor method to predict the evolution of epidemic trends for many regions simultaneously. We construct a 3-way spatio-temporal tensor (location, attribute, time) of case counts and propose a nonnegative tensor factorization with latent epidemiological model regularization named STELAR. Unlike standard tensor factorization methods which cannot predict slabs ahead, STELAR enables long-term prediction by incorporating latent temporal regularization through a system of discrete-time difference equations of a widely adopted epidemiological model. We use latent instead of location/attribute-level epidemiological dynamics to capture common epidemic profile sub-types and improve collaborative learning and prediction. We conduct experiments using both county- and state-level COVID-19 data and show that our model can identify interesting latent patterns of the epidemic. Finally, we evaluate the predictive ability of our method and show superior performance compared to the baselines, achieving up to 21% lower root mean square error and 25% lower mean absolute error for county-level prediction.
Nikos Kargas, Cheng Qian 0001, Nicholas D. Sidiropoulos, Cao Xiao, Lucas Glass, Jimeng Sun 0001
AAAI4
2021 CUP: Cluster Pruning for Compressing Deep Neural Networks
abstract
We propose CUP, a new method for compressing and accelerating deep neural networks. At its core, CUP achieves compression by clustering and pruning similar filters in each layer. For clustering, CUP uses hierarchical clustering which allows for an elegant parameterization of model capacity through a single hyper-parameter t. We observe that by increasing t, CUP can dynamically reduce model capacity through non-uniform layer-wise pruning leading to two advantages. First, CUP can effectively compress a model to within the desired compute budget through a simple line-search on t. Second, through a simple extension, CUP can obtain the pruned model in a single training pass leading to large savings in training time. On Imagenet, CUP leads to a 2.47× FLOPS reduction on Resnet-50 with less than 1% drop in top-5 accuracy. Notably, in the retrain-free setting, CUP-RF saves over 10 hours of training time on 3 GPUs, in comparison to state-of-the-art methods. The code for CUP is open sourced1.
Rahul Duggal, Cao Xiao, Richard W. Vuduc, Polo Chau, Jimeng Sun 0001
IEEE BigData2
2021 MedRetriever: Target-Driven Interpretable Health Risk Prediction via Retrieving Unstructured Medical Text
abstract
The broad adoption of electronic health record (EHR) systems and the advances of deep learning technology have motivated the development of health risk prediction models, which mainly depend on the expressiveness and temporal modeling capacity of deep neural networks (DNNs) to improve prediction performance. Some further augment the prediction by using external knowledge, however, a great deal of EHR information inevitably loses during the knowledge mapping. In addition, prediction made by existing models usually lacks reliable interpretation, which undermines their reliability in guiding clinical decision-making. To solve these challenges, we propose MedRetriever, an effective and flexible framework that leverages unstructured medical text collected from authoritative websites to augment health risk prediction as well as to provide understandable interpretation. Besides, MedRetriever explicitly takes the target disease documents into consideration, which provide key guidance for the model to learn in a target-driven direction, i.e., from the target disease to the input EHR. To specify, MedRetriever can flexibly choose its backbone from major predictive models to learn the EHR embedding for each visit. After that, the EHR embedding and features of target disease documents are aggregated into a query by self-attention to retrieve highly relevant text segments from the medical text pool, which is stored in the dynamically updated text memory. Finally, the comprehensive EHR embedding and the text memory are used for prediction and interpretation. We evaluate MedRetriever against nine state-of-the-art approaches across three real-world EHR datasets, which consistently achieves the best performance in AUC and recall metrics and outperforms the best baseline by at least 4.8% in recall on three test datasets. Furthermore, we conduct case studies to show the easy-to-understand interpretation by MedRetriever.
Muchao Ye, Suhan Cui, Yaqing Wang 0001, Junyu Luo 0001, Cao Xiao, Fenglong Ma
CIKM5
2021 Multi-version Tensor Completion for Time-delayed Spatio-temporal Data
abstract
Real-world spatio-temporal data is often incomplete or inaccurate due to various data loading delays. For example, a location-disease-time tensor of case counts can have multiple delayed updates of recent temporal slices for some locations or diseases. Recovering such missing or noisy (under-reported) elements of the input tensor can be viewed as a generalized tensor completion problem. Existing tensor completion methods usually assume that i) missing elements are randomly distributed and ii) noise for each tensor element is i.i.d. zero-mean. Both assumptions can be violated for spatio-temporal tensor data. We often observe multiple versions of the input tensor with different under-reporting noise levels. The amount of noise can be time- or location-dependent as more updates are progressively introduced to the tensor. We model such dynamic data as a multi-version tensor with an extra tensor mode capturing the data updates. We propose a low-rank tensor model to predict the updates over time. We demonstrate that our method can accurately predict the ground-truth values of many real-world tensors. We obtain up to 27.2% lower root mean-squared-error compared to the best baseline method. Finally, we extend our method to track the tensor data over time, leading to significant computational savings.
Cheng Qian 0001, Nikos Kargas, Cao Xiao, Lucas Glass, Nicholas D. Sidiropoulos, Jimeng Sun 0001
IJCAI3
2021 Change Matters: Medication Change Prediction with Recurrent Residual Networks
abstract
Deep learning is revolutionizing predictive healthcare, including recommending medications to patients with complex health conditions. Existing approaches focus on predicting all medications for the current visit, which often overlaps with medications from previous visits. A more clinically relevant task is to identify medication changes. In this paper, we propose a new recurrent residual networks, named MICRON, for medication change prediction. MICRON takes the changes in patient health records as input and learns to update a hid- den medication vector and the medication set recurrently with a reconstruction design. The medication vector is like the memory cell that encodes longitudinal information of medications. Unlike traditional methods that require the entire patient history for prediction, MICRON has a residual-based inference that allows for sequential updating based only on new patient features (e.g., new diagnoses in the recent visit), which is efficient. We evaluated MICRON on real inpatient and outpatient datasets. MICRON achieves 3.5% and 7.8% relative improvements over the best baseline in F1 score, respectively. MICRON also requires fewer parameters, which significantly reduces the training time to 38.3s per epoch with 1.5× speed-up.
Chaoqi Yang, Cao Xiao, Lucas Glass, Jimeng Sun 0001
IJCAI2
2021 SafeDrug: Dual Molecular Graph Encoders for Recommending Effective and Safe Drug Combinations
abstract
Medication recommendation is an essential task of AI for healthcare. Existing works focused on recommending drug combinations for patients with complex health conditions solely based on their electronic health records. Thus, they have the following limitations: (1) some important data such as drug molecule structures have not been utilized in the recommendation process. (2) drug-drug interactions (DDI) are modeled implicitly, which can lead to sub-optimal results. To address these limitations, we propose a DDI-controllable drug recommendation model named SafeDrug to leverage drugs’ molecule structures and model DDIs explicitly. SafeDrug is equipped with a global message passing neural network (MPNN) module and a local bipartite learning module to fully encode the connectivity and functionality of drug molecules. SafeDrug also has a controllable loss function to control DDI level in the recommended drug combinations effectively. On a benchmark dataset, our SafeDrug is relatively shown to reduce DDI by 19.43% and improves 2.88% on Jaccard similarity between recommended and actually prescribed drug combinations over previous approaches. Moreover, SafeDrug also requires much fewer parameters than previous deep learning based approaches, leading to faster training by about 14% and around 2× speed-up in inference.
Chaoqi Yang, Cao Xiao, Fenglong Ma, Lucas Glass, Jimeng Sun 0001
IJCAI2
2021 Probabilistic and Dynamic Molecule-Disease Interaction Modeling for Drug Discovery
abstract
Drug discovery aims at finding promising drug molecules for treating target diseases. Existing computational drug discovery methods mainly depend on molecule databases, ignoring valuable data collected from clinical trials. In this work, we propose PRIME to leverage high-quality drug molecules and drug-disease relations in historical clinical trials to narrow down the molecular search space in drug discovery. PRIME also introduces time dependency constraints to model evolving drug-disease relations using a probabilistic deep learning model that can quantify model uncertainty. We evaluated PRIME against leading models on both de novo design and drug repurposing tasks. Results show that compared with the best baselines, PRIME achieves 25.9% relative improvement (i.e., reduction) in average hit-ranking on drug repurposing and 47.6% relative improvement in success rate on de novo design.
Tianfan Fu, Cao Xiao, Cheng Qian 0001, Lucas Glass, Jimeng Sun 0001
KDD2
2021 Advances in Mining Heterogeneous Healthcare Data
abstract
Thanks to the explosion of heterogeneous healthcare data and advanced machine learning and data mining techniques, specifically deep learning methods, we now have an opportunity to make difference in healthcare. In this tutorial, we will present state-of-the-art deep learning methods and their real-world applications, specifically focusing on exploring the unique characteristics of different types of healthcare data. The first half will be spent on introducing recent advances in mining structured healthcare data, including computational phenotyping, disease early detection/risk prediction and treatment recommendation. In the second half, we will focus on challenges specific to the unstructured healthcare data, and introduce advanced deep learning methods in automated ICD coding, understandable medical language translation, clinical trial mining, and medical report generation. This tutorial is intended for students, engineers and researchers who are interested in applying deep learning methods to healthcare, and prerequisite knowledge will be minimal. The tutorial will be concluded with open problems and a Q&A session.
Fenglong Ma, Muchao Ye, Junyu Luo 0001, Cao Xiao, Jimeng Sun 0001
KDD4
2021 MTC: Multiresolution Tensor Completion from Partial and Coarse Observations
abstract
Existing tensor completion formulation mostly relies on partial observations from a single tensor. However, tensors extracted from real-world data often are more complex due to: (i) Partial observation: Only a small subset of tensor elements are available. (ii) Coarse observation: Some tensor modes only present coarse and aggregated patterns (e.g., monthly summary instead of daily reports). In this paper, we are given a subset of the tensor and some aggregated/coarse observations (along one or more modes) and seek to recover the original fine-granular tensor with low-rank factorization. We formulate a coupled tensor completion problem and propose an efficient Multi-resolution Tensor Completion model (MTC) to solve the problem. Our MTC model explores tensor mode properties and leverages the hierarchy of resolutions to recursively initialize an optimization setup, and optimizes on the coupled system using alternating least squares. MTC ensures low computational and space complexity. We evaluate our model on two COVID-19 related spatio-temporal tensors. The experiments show that MTC could provide 65.20% and 75.79% percentage of fitness (PoF) in tensor completion with only 5% fine granular observations, which is 27.96% relative improvement over the best baseline. To evaluate the learned low-rank factors, we also design a tensor prediction task for daily and cumulative disease case predictions, where MTC achieves 50% in PoF and 30% relative improvements over the best baseline.
Chaoqi Yang, Cao Xiao, Cheng Qian 0001, Edgar Solomonik, Jimeng Sun 0001
KDD3
2021 UNITE: Uncertainty-based Health Risk Prediction Leveraging Multi-sourced Data
abstract
Successful health risk prediction demands accuracy and reliability of the model. Existing predictive models mainly depend on mining electronic health records (EHR) with advanced deep learning techniques to improve model accuracy. However, they all ignore the importance of publicly available online health data, especially socioeconomic status, environmental factors, and detailed demographic information for each location, which are all strong predictive signals and can definitely augment precision medicine. To achieve model reliability, the model needs to provide accurate prediction and uncertainty score of the prediction. However, existing uncertainty estimation approaches often failed in handling high-dimensional data, which are present in multi-sourced data.
Chacha Chen, Fenglong Ma, Lucas Glass, Jimeng Sun 0001, Cao Xiao
WWW6
2021 AID: Active Distillation Machine to Leverage Pre-Trained Black-Box Models in Private Data Settings
abstract
This paper presents an active distillation method for a local institution (e.g., hospital) to find the best queries within its given budget to distill an on-server black-box model’s predictive knowledge into a local surrogate with transparent parameterization. This allows local institutions to understand better the predictive reasoning of the black-box model in its own local context or to further customize the distilled knowledge with its private dataset that cannot be centralized and fed into the server model. The proposed method thus addresses several challenges of deploying machine learning (ML) in many industrial settings (e.g., healthcare analytics) with strong proprietary constraints. These include: (1) the opaqueness of the server model’s architecture which prevents local users from understanding its predictive reasoning in their local data contexts; (2) the increasing cost and risk of uploading local data on the cloud for analysis; and (3) the need to customize the server model with private onsite data. We evaluated the proposed method on both benchmark and real-world healthcare data where significant improvements over existing local distillation methods were observed. A theoretical analysis of the proposed method is also presented.
Trong Nghia Hoang, Shenda Hong, Cao Xiao, Bryan Low, Jimeng Sun 0001
WWW3
2021 MedPath: Augmenting Health Risk Prediction via Medical Knowledge Paths
abstract
The broad adoption of electronic health records (EHR) data and the availability of biomedical knowledge graphs (KGs) on the web have provided clinicians and researchers unprecedented resources and opportunities for conducting health risk predictions to improve healthcare quality and medical resource allocation. Existing methods have focused on improving the EHR feature representations using attention mechanisms, time-aware models, or external knowledge. However, they ignore the importance of using personalized information to make predictions. Besides, the reliability of their prediction interpretations needs to be improved since their interpretable attention scores are not explicitly reasoned from disease progression paths. In this paper, we propose MedPath to solve these challenges and augment existing risk prediction models with the ability to use personalized information and provide reliable interpretations inferring from disease progression paths. Firstly, MedPath extracts personalized knowledge graphs (PKGs) containing all possible disease progression paths from observed symptoms to target diseases from a large-scale online medical knowledge graph. Next, to augment existing EHR encoders for achieving better predictions, MedPath learns a PKG embedding by conducting multi-hop message passing from symptom nodes to target disease nodes through a graph neural network encoder. Since MedPath reasons disease progression by paths existing in PKGs, it can provide explicit explanations for the prediction by pointing out how observed symptoms can finally lead to target diseases. Experimental results on three real-world medical datasets show that MedPath is effective in improving the performance of eight state-of-the-art methods with higher F1 scores and AUCs. Our case study also demonstrates that MedPath can greatly improve the explicitness of the risk prediction interpretation.1
Muchao Ye, Suhan Cui, Yaqing Wang 0001, Junyu Luo 0001, Cao Xiao, Fenglong Ma
WWW5
2021 DeepPurpose: a deep learning library for drug-target interaction prediction
abstract
SUMMARY: Accurate prediction of drug-target interactions (DTI) is crucial for drug discovery. Recently, deep learning (DL) models for show promising performance for DTI prediction. However, these models can be difficult to use for both computer scientists entering the biomedical field and bioinformaticians with limited DL experience. We present DeepPurpose, a comprehensive and easy-to-use DL library for DTI prediction. DeepPurpose supports training of customized DTI prediction models by implementing 15 compound and protein encoders and over 50 neural architectures, along with providing many other useful features. We demonstrate state-of-the-art performance of DeepPurpose on several benchmark datasets. AVAILABILITY AND IMPLEMENTATION: https://github.com/kexinhuang12345/DeepPurpose. SUPPLEMENTARY INFORMATION: Supplementary data are available at Bioinformatics online.
Tianfan Fu, Lucas Glass, Marinka Zitnik, Cao Xiao, Jimeng Sun 0001
Bioinform.5
2021 MolTrans: Molecular Interaction Transformer for drug-target interaction prediction
abstract
MOTIVATION: Drug-target interaction (DTI) prediction is a foundational task for in-silico drug discovery, which is costly and time-consuming due to the need of experimental search over large drug compound space. Recent years have witnessed promising progress for deep learning in DTI predictions. However, the following challenges are still open: (i) existing molecular representation learning approaches ignore the sub-structural nature of DTI, thus produce results that are less accurate and difficult to explain and (ii) existing methods focus on limited labeled data while ignoring the value of massive unlabeled molecular data. RESULTS: We propose a Molecular Interaction Transformer (MolTrans) to address these limitations via: (i) knowledge inspired sub-structural pattern mining algorithm and interaction modeling module for more accurate and interpretable DTI prediction and (ii) an augmented transformer encoder to better extract and capture the semantic relations among sub-structures extracted from massive unlabeled biomedical data. We evaluate MolTrans on real-world data and show it improved DTI prediction performance compared to state-of-the-art baselines. AVAILABILITY AND IMPLEMENTATION: The model scripts are available at https://github.com/kexinhuang12345/moltrans. SUPPLEMENTARY INFORMATION: Supplementary data are available at Bioinformatics online.
Cao Xiao, Lucas Glass, Jimeng Sun 0001
Bioinform.2
2021 SumGNN: multi-typed drug interaction prediction via efficient knowledge graph summarization
abstract
MOTIVATION: Thanks to the increasing availability of drug-drug interactions (DDI) datasets and large biomedical knowledge graphs (KGs), accurate detection of adverse DDI using machine learning models becomes possible. However, it remains largely an open problem how to effectively utilize large and noisy biomedical KG for DDI detection. Due to its sheer size and amount of noise in KGs, it is often less beneficial to directly integrate KGs with other smaller but higher quality data (e.g. experimental data). Most of existing approaches ignore KGs altogether. Some tries to directly integrate KGs with other data via graph neural networks with limited success. Furthermore most previous works focus on binary DDI prediction whereas the multi-typed DDI pharmacological effect prediction is more meaningful but harder task. RESULTS: To fill the gaps, we propose a new method SumGNN: knowledge summarization graph neural network, which is enabled by a subgraph extraction module that can efficiently anchor on relevant subgraphs from a KG, a self-attention based subgraph summarization scheme to generate reasoning path within the subgraph, and a multi-channel knowledge and data integration module that utilizes massive external biomedical knowledge for significantly improved multi-typed DDI predictions. SumGNN outperforms the best baseline by up to 5.54%, and performance gain is particularly significant in low data relation types. In addition, SumGNN provides interpretable prediction via the generated reasoning paths for each prediction. AVAILABILITY AND IMPLEMENTATION: The code is available in Supplementary Material. SUPPLEMENTARY INFORMATION: Supplementary data are available at Bioinformatics online.
Yue Yu 0001, Chao Zhang 0014, Lucas Glass, Jimeng Sun 0001, Cao Xiao
Bioinform.6
2021 STAN: spatio-temporal attention network for pandemic prediction using real-world evidence
abstract
OBJECTIVE: The COVID-19 pandemic has created many challenges that need immediate attention. Various epidemiological and deep learning models have been developed to predict the COVID-19 outbreak, but all have limitations that affect the accuracy and robustness of the predictions. Our method aims at addressing these limitations and making earlier and more accurate pandemic outbreak predictions by (1) using patients' EHR data from different counties and states that encode local disease status and medical resource utilization condition; (2) considering demographic similarity and geographical proximity between locations; and (3) integrating pandemic transmission dynamics into deep learning models. MATERIALS AND METHODS: We proposed a spatio-temporal attention network (STAN) for pandemic prediction. It uses an attention-based graph convolutional network to capture geographical and temporal trends and predict the number of cases for a fixed number of days into the future. We also designed a physical law-based loss term for enhancing long-term prediction. STAN was tested using both massive real-world patient data and open source COVID-19 statistics provided by Johns Hopkins university across all U.S. counties. RESULTS: STAN outperforms epidemiological modeling methods such as SIR and SEIR and deep learning models on both long-term and short-term predictions, achieving up to 87% lower mean squared error compared to the best baseline prediction model. CONCLUSIONS: By using information from real-world patient data and geographical data, STAN can better capture the disease status and medical resource utilization information and thus provides more accurate pandemic modeling. With pandemic transmission law based regularization, STAN also achieves good long-term prediction performance.
Rakshith Sharma Srinivasa, Cheng Qian 0001, Lucas Glass, Jeffrey Spaeder, Justin K. Romberg, Jimeng Sun 0001, Cao Xiao
J. Am. Medical Informatics Assoc.8
2021 FLANNEL (Focal Loss bAsed Neural Network EnsembLe) for COVID-19 detection
abstract
OBJECTIVE: The study sought to test the possibility of differentiating chest x-ray images of coronavirus disease 2019 (COVID-19) against other pneumonia and healthy patients using deep neural networks. MATERIALS AND METHODS: We construct the radiography (x-ray) imaging data from 2 publicly available sources, which include 5508 chest x-ray images across 2874 patients with 4 classes: normal, bacterial pneumonia, non-COVID-19 viral pneumonia, and COVID-19. To identify COVID-19, we propose a FLANNEL (Focal Loss bAsed Neural Network EnsembLe) model, a flexible module to ensemble several convolutional neural network models and fuse with a focal loss for accurate COVID-19 detection on class imbalance data. RESULTS: FLANNEL consistently outperforms baseline models on COVID-19 identification task in all metrics. Compared with the best baseline, FLANNEL shows a higher macro-F1 score, with 6% relative increase on the COVID-19 identification task, in which it achieves precision of 0.7833 ± 0.07, recall of 0.8609 ± 0.03, and F1 score of 0.8168 ± 0.03. DISCUSSION: Ensemble learning that combines multiple independent basis classifiers can increase the robustness and accuracy. We propose a neural weighing module to learn the importance weight for each base model and combine them via weighted ensemble to get the final classification results. In order to handle the class imbalance challenge, we adapt focal loss to our multiple classification task as the loss function. CONCLUSION: FLANNEL effectively combines state-of-the-art convolutional neural network classification models and tackles class imbalance with focal loss to achieve better performance on COVID-19 detection from x-rays.
Zhi Qiao 0007, Austin Bae, Lucas Glass, Cao Xiao, Jimeng Sun 0001
J. Am. Medical Informatics Assoc.4
2021 An Adaptive Pattern Learning Framework to Personalize Online Seizure Prediction
abstract
The sudden and spontaneous occurrence of epileptic seizures can impose a significant burden on patients with epilepsy. If seizure onset can be prospectively predicted, it could greatly improve the life of patients with epilepsy and also open new therapeutic avenues for epilepsy treatment. However, discovering effective predictive patterns from massive brainwave signals is still a challenging problem. The prediction of epileptic seizures is still in its early stage. Most existing studies actually investigated the predictability of seizures offline instead of a truly prospective online prediction, and also the high inter-individual variability was not fully considered in prediction. In this study, we propose a novel adaptive pattern learning framework with a new online feature extraction approach to achieve personalized online prospective seizure prediction. In particular, a two-level online feature extraction approach is applied to monitor intracranial electroencephalogram (EEG) signals and construct a pattern library incrementally. Three prediction rules were developed and evaluated based on the continuously-updated patient-specific pattern library for each patient, including the adaptive probabilistic prediction (APP), adaptive linear-discriminant-analysis-based prediction (ALP), and adaptive Naive Bayes-based prediction (ANBP). The proposed online pattern learning and prediction system achieved impressive prediction results for 10 patients with epilepsy using long-term EEG recordings. The best testing prediction accuracy averaged over the 10 patients were 79, 78, and 82 percent for the APP, ALP, and ANBP prediction scheme, respectively. The experimental results confirmed that the proposed adaptive prediction framework offers a promising practical tool to solve the challenging seizure prediction problem.
Cao Xiao, Leonidas D. Iasemidis, W. Art Chaovalitwongse
IEEE Trans. Big Data1
2020 Doctor2Vec: Dynamic Doctor Representation Learning for Clinical Trial Recruitment
abstract
Massive electronic health records (EHRs) enable the success of learning accurate patient representations to support various predictive health applications. In contrast, doctor representation was not well studied despite that doctors play pivotal roles in healthcare. How to construct the right doctor representations? How to use doctor representation to solve important health analytic problems? In this work, we study the problem on clinical trial recruitment, which is about identifying the right doctors to help conduct the trials based on the trial description and patient EHR data of those doctors. We propose Doctor2Vec which simultaneously learns 1) doctor representations from EHR data and 2) trial representations from the description and categorical information about the trials. In particular, Doctor2Vec utilizes a dynamic memory network where the doctor's experience with patients are stored in the memory bank and the network will dynamically assign weights based on the trial representation via an attention mechanism. Validated on large real-world trials and EHR data including 2,609 trials, 25K doctors and 430K patients, Doctor2Vec demonstrated improved performance over the best baseline by up to 8.7% in PR-AUC. We also demonstrated that the Doctor2Vec embedding can be transferred to benefit data insufficiency settings including trial recruitment in less populated/newly explored country with 13.7% improvement or for rare diseases with 8.1% improvement in PR-AUC.
Siddharth Biswal, Cao Xiao, Lucas Glass, Elizabeth Milkovits, Jimeng Sun 0001
AAAI2
2020 CONAN: Complementary Pattern Augmentation for Rare Disease Detection
abstract
Rare diseases affect hundreds of millions of people worldwide but are hard to detect since they have extremely low prevalence rates (varying from 1/1,000 to 1/200,000 patients) and are massively underdiagnosed. How do we reliably detect rare diseases with such low prevalence rates? How to further leverage patients with possibly uncertain diagnosis to improve detection? In this paper, we propose a Complementary pattern Augmentation (CONAN) framework for rare disease detection. CONAN combines ideas from both adversarial training and max-margin classification. It first learns self-attentive and hierarchical embedding for patient pattern characterization. Then, we develop a complementary generative adversarial networks (GAN) model to generate candidate positive and negative samples from the uncertain patients by encouraging a max-margin between classes. In addition, CONAN has a disease detector that serves as the discriminator during the adversarial training for identifying rare diseases. We evaluated CONAN on two disease detection tasks. For low prevalence inflammatory bowel disease (IBD) detection, CONAN achieved .96 precision recall area under the curve (PR-AUC) and 50.1% relative improvement over the best baseline. For rare disease idiopathic pulmonary fibrosis (IPF) detection, CONAN achieves .22 PR-AUC with 41.3% relative improvement over the best baseline.
Limeng Cui, Siddharth Biswal, Lucas Glass, Greg Lever, Jimeng Sun 0001, Cao Xiao
AAAI6
2020 CORE: Automatic Molecule Optimization Using Copy & Refine Strategy
abstract
Molecule optimization is about generating molecule Y with more desirable properties based on an input molecule X. The state-of-the-art approaches partition the molecules into a large set of substructures S and grow the new molecule structure by iteratively predicting which substructure from S to add. However, since the set of available substructures S is large, such an iterative prediction task is often inaccurate especially for substructures that are infrequent in the training data. To address this challenge, we propose a new generating strategy called “Copy&Refine” (CORE), where at each step the generator first decides whether to copy an existing substructure from input X or to generate a new substructure, then the most promising substructure will be added to the new molecule. Combining together with scaffolding tree generation and adversarial training, CORE can significantly improve several latest molecule optimization methods in various measures including drug likeness (QED), dopamine receptor (DRD2) and penalized LogP. We tested CORE and baselines using the ZINC database and CORE obtained up to 11% and 21% relatively improvement over the baselines on success rate on the complete test set and the subset with infrequent substructures, respectively.
Tianfan Fu, Cao Xiao, Jimeng Sun 0001
AAAI2
2020 CASTER: Predicting Drug Interactions with Chemical Substructure Representation
abstract
Adverse drug-drug interactions (DDIs) remain a leading cause of morbidity and mortality. Identifying potential DDIs during the drug design process is critical for patients and society. Although several computational models have been proposed for DDI prediction, there are still limitations: (1) specialized design of drug representation for DDI predictions is lacking; (2) predictions are based on limited labelled data and do not generalize well to unseen drugs or DDIs; and (3) models are characterized by a large number of parameters, thus are hard to interpret. In this work, we develop a ChemicAl SubstrucTurE Representation (CASTER) framework that predicts DDIs given chemical structures of drugs. CASTER aims to mitigate these limitations via (1) a sequential pattern mining module rooted in the DDI mechanism to efficiently characterize functional sub-structures of drugs; (2) an auto-encoding module that leverages both labelled and unlabelled chemical structure data to improve predictive accuracy and generalizability; and (3) a dictionary learning module that explains the prediction via a small set of coefficients which measure the relevance of each input sub-structures to the DDI outcome. We evaluated CASTER on two real-world DDI datasets and showed that it performed better than state-of-the-art baselines and provided interpretable predictions.
Cao Xiao, Trong Nghia Hoang, Lucas Glass, Jimeng Sun 0001
AAAI2
2020 α-MOP: Molecule optimization with α-divergence
abstract
Automatic molecule optimization aims at generating a new molecule Y with more desirable properties based on an input molecule X. There are two learning strategies for molecule optimization: 1) Maximum Likelihood Estimation (MLE) methods that take a set of molecule pairs (X, Y ) as training data where X is the input molecule and Y is the enhanced molecule. However, such molecule pairs are not naturally available in molecule databases and has to be constructed using ad hoc heuristics, which limits the performance of MLE methods. 2) Reinforcement Learning (RL) methods, though bypass the need of molecule pairs as training data, suffer from poor exploration efficiency, especially in the early phase of learning. To address both challenges, we propose α-Molecular OPtimization (α-MOP), which uses α-divergence to unify both MLE and RL objectives automatically. In early phase it focuses more on maximum likelihood objective but gradually shifts more weight onto reinforcement learning objective. Evaluated on multiple datasets, α-MOP obtains success rate of 49.91% in QED, 49.32% in DRD2 and 56.43% in LogP, which outperforms both MLE and RL based molecule optimization methods.
Tianfan Fu, Cao Xiao, Lucas Glass, Jimeng Sun 0001
BIBM2
2020 Supervised Topic Compositional Neural Language Model for Clinical Narrative Understanding
abstract
Clinical narratives that describe complex medical events are often accompanied by meta-information such as a patient’s demographics, diagnoses and medications. This structured information implicitly relates to the logical and semantic structure of the entire narrative, and thus affects vocabulary choices for the narrative composition. To leverage this meta-information, we propose a supervised topic compositional neural language model, called MeTRNN, that integrates the strength of supervised topic modeling in capturing global semantics with the capacity of contextual recurrent neural networks (RNN) in modeling local word dependencies. MeTRNN generates interpretable topics from global meta-information and uses them to facilitate contextual RNNs in modeling local dependencies of text. For efficient training of MeTRNN, we develop an autoencoding variational Bayes inference method. We evaluate MeTRNN on the word prediction tasks using public text datasets. MeTRNN consistently outperforms all baselines across all datasets in perplexity ranging from 5% to 40%. Our case studies on real world electronic health records (EHR) data show that MeTRNN can learn and benefit from meaningful topics.
Xiao Qin 0003, Cao Xiao, Tengfei Ma 0001, Tabassum Kakar, Susmitha Wunnava, Xiangnan Kong, Elke A. Rundensteiner, Fei Wang 0001
IEEE BigData2
2020 LSAN: Modeling Long-term Dependencies and Short-term Correlations with Hierarchical Attention for Risk Prediction
abstract
Risk prediction using electronic health records (EHR) is a challenging data mining task due to the two-level hierarchical structure of EHR data. EHR data consist of a set of time-ordered visits, and within each visit, there is a set of unordered diagnosis codes. Existing approaches focus on modeling temporal visits with deep neural network (DNN) techniques. However, they ignore the importance of modeling diagnosis codes within visits, and a lot of task-unrelated information within visits usually leads to unsatisfactory performance of existing approaches. To minimize the effect caused by noise information of EHR data, in this paper, we propose a novel DNN for risk prediction termed as LSAN, which consists of a Hierarchical Attention Module (HAM) and a Temporal Aggregation Module (TAM). Particularly, LSAN applies HAM to model the hierarchical structure of EHR data. Using the attention mechanism in the hierarchy of diagnosis code, HAM is able to retain diagnosis details and assign flexible attention weights to different diagnosis codes by their relevance to corresponding diseases. Moreover, the attention mechanism in the hierarchy of visit learns a comprehensive feature throughout the visit history by paying greater attention to visits with higher relevance. Based on the foundation laying by HAM, TAM uses a two-pathway structure to learn a robust temporal aggregation mechanism among all visits for LSAN. It extracts long-term dependencies by a Transformer encoder and short-term correlations by a parallel convolutional layer among different visits. With the construction of HAM and TAM, LSAN achieves the state-of-the-art performance on three real-world datasets with larger AUCs, recalls and F1 scores. Furthermore, the model analysis results demonstrate the effectiveness of the network construction with good interpretability and robustness of decision making by LSAN.
Muchao Ye, Junyu Luo 0001, Cao Xiao, Fenglong Ma
CIKM3
2020 COMPOSE: Cross-Modal Pseudo-Siamese Network for Patient Trial Matching
abstract
Clinical trials play important roles in drug development but often suffer from expensive, inaccurate and insufficient patient recruitment. The availability of massive electronic health records (EHR) data and trial eligibility criteria (EC) bring a new opportunity to data driven patient recruitment. One key task named patient-trial matching is to find qualified patients for clinical trials given structured EHR and unstructured EC text (both inclusion and exclusion criteria). How to match complex EC text with longitudinal patient EHRs? How to embed many-to-many relationships between patients and trials? How to explicitly handle the difference between inclusion and exclusion criteria? In this paper, we proposed CrOss-Modal PseudO-SiamEse network (COMPOSE) to address these challenges for patient-trial matching. One path of the network encodes EC using convolutional highway network. The other path processes EHR with multi-granularity memory network that encodes structured patient records into multiple levels based on medical ontology. Using the EC embedding as query, COMPOSE performs attentional record alignment and thus enables dynamic patient-trial matching. COMPOSE also introduces a composite loss term to maximize the similarity between patient records and inclusion criteria while minimize the similarity to the exclusion criteria. Experiment results show COMPOSE can reach 98.0% AUC on patient-criteria matching and 83.7% accuracy on patient-trial matching, which leads 24.3% improvement over the best baseline on real-world patient-trial matching tasks.
Cao Xiao, Lucas Glass, Jimeng Sun 0001
KDD2
2020 HiTANet: Hierarchical Time-Aware Attention Networks for Risk Prediction on Electronic Health Records
abstract
Deep learning methods especially recurrent neural network based models have demonstrated early success in disease risk prediction on longitudinal patient data. Existing works follow a strong assumption to implicitly assume the stationary disease progression during each time period, and thus, take a homogeneous way to decay the information from previous time steps for all patients. However,in reality, disease progression is non-stationary. Besides, the key time steps for a target disease vary among patients. To leverage time information for risk prediction in a more reasonable way, we propose a new hierarchical time-aware attention network, named HiTANet, which imitates the decision making process of doctors inrisk prediction. Particularly, HiTANet models time information in local and global stages. The local evaluation stage has a time aware Transformer that embeds time information into visit-level embed-ding and generates local attention weight for each visit. The global synthesis stage further adopts a time-aware key-query attention mechanism to assign global weights to different time steps. Finally, the two types of attention weights are dynamically combined to generate the patient representations for further risk prediction. We evaluate HiTANet on three real-world datasets. Compared with the best results among twelve competing baselines, HiTANet achieves over 7% in terms of F1 score on all datasets, which demonstrates the effectiveness of the proposed model and the necessity of modeling time information in risk prediction task.
Junyu Luo 0001, Muchao Ye, Cao Xiao, Fenglong Ma
KDD3
2020 Clinical Report Auto-completion
abstract
Generating clinical reports from raw recordings such as X-rays and electroencephalogram (EEG) is an essential and routine task for doctors. However, it is often time-consuming to write accurate and detailed reports. Most existing methods try to generate the whole reports from the raw input with limited success because 1) generated reports often contain errors that need manual review and correction, 2) it does not save time when doctors want to write additional information into the report, and 3) the generated reports are not customized based on individual doctors’ preference. We propose CLinicAl Report Auto-completion (, an interactive method that generates reports in a sentence by sentence fashion based on doctors’ anchor words and partially completed sentences. earches for most relevant sentences from existing reports as the template for the current report. The retrieved sentences are sequentially modified by combining with the input feature representations to create the final report. In our experimental evaluation chieved 0.393 CIDEr and 0.248 BLEU-4 on X-ray reports and 0.482 CIDEr and 0.491 BLEU-4 for EEG reports for sentence-level generation, which is up to 35% improvement over the best baseline. Also via our qualitative evaluation, s shown to produce reports which have a significantly higher level of approval by doctors in a user study (3.74 out of 5 for s 2.52 out of 5 for the baseline).
Siddharth Biswal, Cao Xiao, Lucas Glass, M. Brandon Westover, Jimeng Sun 0001
WWW2
2020 REST: Robust and Efficient Neural Networks for Sleep Monitoring in the Wild
abstract
In recent years, significant attention has been devoted towards integrating deep learning technologies in the healthcare domain. However, to safely and practically deploy deep learning models for home health monitoring, two significant challenges must be addressed: the models should be (1) robust against noise; and (2) compact and energy-efficient. We propose Rest , a new method that simultaneously tackles both issues via 1) adversarial training and controlling the Lipschitz constant of the neural network through spectral regularization while 2) enabling neural network compression through sparsity regularization. We demonstrate that Rest produces highly-robust and efficient models that substantially outperform the original full-sized models in the presence of noise. For the sleep staging task over single-channel electroencephalogram (EEG), the Rest model achieves a macro-F1 score of 0.67 vs. 0.39 achieved by a state-of-the-art model in the presence of Gaussian noise while obtaining 19 × parameter reduction and 15 × MFLOPS reduction on two large, real-world EEG datasets. By deploying these models to an Android application on a smartphone, we quantitatively observe that Rest allows models to achieve up to 17 × energy reduction and 9 × faster inference. We open source the code repository with this paper: https://github.com/duggalrahul/REST.
Rahul Duggal, Scott Freitas, Cao Xiao, Polo Chau, Jimeng Sun 0001
WWW3
2020 StageNet: Stage-Aware Neural Networks for Health Risk Prediction
abstract
Deep learning has demonstrated success in health risk prediction especially for patients with chronic and progressing conditions. Most existing works focus on learning disease patterns from longitudinal patient data, but pay little attention to the disease progression stage itself. To fill the gap, we propose a Stage-aware neural Network (StageNet) model to extract disease stage information from patient data and integrate it into risk prediction. StageNet is enabled by (1) a stage-aware long short-term memory (LSTM) module that extracts health stage variations unsupervisedly; (2) a stage-adaptive convolutional module that incorporates stage-related progression patterns into risk prediction. We evaluate StageNet on two real-world datasets and show that StageNet outperforms state-of-the-art models in risk prediction task and patient subtyping task. Compared to the best baseline model, StageNet achieves up to 12% higher AUPRC for risk prediction task on two real-world patient datasets. StageNet also achieves over 58% higher Calinski-Harabasz score (a cluster quality metric) for a patient subtyping task.
Cao Xiao, Yasha Wang, Wen Tang 0001, Lucas Glass, Jimeng Sun 0001
WWW2
2020 Patient-Trial Matching with Deep Embedding and Entailment Prediction
abstract
Clinical trials are essential for drug development but often suffer from expensive, inaccurate and insufficient patient recruitment. The core problem of patient-trial matching is to find qualified patients for a trial, where patient information is stored in electronic health records (EHR) while trial eligibility criteria (EC) are described in text documents available on the web. How to represent longitudinal patient EHR? How to extract complex logical rules from EC? Most existing works rely on manual rule-based extraction, which is time consuming and inflexible for complex inference. To address these challenges, we proposed a cross-modal inference learning model to jointly encode enrollment criteria (text) and patients records (tabular data) into a shared latent space for matching inference. pplies a pre-trained Bidirectional Encoder Representations from Transformers(BERT) model to encode clinical trial information into sentence embedding. And uses a hierarchical embedding model to represent patient longitudinal EHR. In addition, s augmented by a numerical information embedding and entailment module to reason over numerical information in both EC and EHR. These encoders are trained jointly to optimize patient-trial matching score. We evaluated n the trial-patient matching task with demonstrated on real world datasets. utperformed the best baseline by up to 12.4% in average F1.
Cao Xiao, Lucas Glass, Jimeng Sun 0001
WWW2
2020 Dr. Agent: Clinical predictive model via mimicked second opinions
abstract
OBJECTIVE: Prediction of disease phenotypes and their outcomes is a difficult task. In practice, patients routinely seek second opinions from multiple clinical experts for complex disease diagnosis. Our objective is to mimic such a practice of seeking second opinions by training 2 agents with different focuses: the primary agent studies the most recent visit of the patient to learn the current health status, and then the second-opinion agent considers the entire patient history to obtain a more global view. MATERIALS AND METHODS: Our approach Dr. Agent augments recurrent neural networks with 2 policy gradient agents. Moreover, Dr. Agent is customized with various patient demographics information and learns a dynamic skip connection to focus on the relevant information over time. We trained Dr. Agent to perform 4 clinical prediction tasks on the publicly available MIMIC-III (Medical Information Mart for Intensive Care) database: (1) in-hospital mortality prediction, (2) acute care phenotype classification, (3) physiologic decompensation prediction, and (4) forecasting length of stay. We compared the performance of Dr. Agent against 4 baseline clinical predictive models. RESULTS: Dr. Agent outperforms baseline clinical prediction models across all 4 tasks in terms of all metrics. Compared with the best baseline model, Dr. Agent achieves up to 15% higher area under the precision-recall curve on different tasks. CONCLUSIONS: Dr. Agent can comprehensively model the long-term dependencies of patients' health status while considering patients' demographics using 2 agents, and therefore achieves better prediction performance on different clinical prediction tasks.
Cao Xiao, Lucas Glass, Jimeng Sun 0001
J. Am. Medical Informatics Assoc.2
2019 GAMENet: Graph Augmented MEmory Networks for Recommending Medication Combination
abstract
Recent progress in deep learning is revolutionizing the healthcare domain including providing solutions to medication recommendations, especially recommending medication combination for patients with complex health conditions. Existing approaches either do not customize based on patient health history, or ignore existing knowledge on drug-drug interactions (DDI) that might lead to adverse outcomes. To fill this gap, we propose the Graph Augmented Memory Networks (GAMENet), which integrates the drug-drug interactions knowledge graph by a memory module implemented as a graph convolutional networks, and models longitudinal patient records as the query. It is trained end-to-end to provide safe and personalized recommendation of medication combination. We demonstrate the effectiveness and safety of GAMENet by comparing with several state-of-the-art methods on real EHR data. GAMENet outperformed all baselines in all effectiveness measures, and also achieved 3.60% DDI rate reduction from existing EHR data.
Junyuan Shang, Cao Xiao, Tengfei Ma 0001, Hongyan Li 0002, Jimeng Sun 0001
AAAI2
2019 DDL: Deep Dictionary Learning for Predictive Phenotyping
abstract
Predictive phenotyping is about accurately predicting what phenotypes will occur in the next clinical visit based on longitudinal Electronic Health Record (EHR) data. While deep learning (DL) models have recently demonstrated strong performance in predictive phenotyping, they require access to a large amount of labeled data, which are expensive to acquire. To address this label-insufficient challenge, we propose a deep dictionary learning framework (DDL) for phenotyping, which utilizes unlabeled data as a complementary source of information to generate a better, more succinct data representation. Our empirical evaluations on multiple EHR datasets demonstrated that DDL outperforms the existing predictive phenotyping methods on a wide variety of clinical tasks that require patient phenotyping. The results also show that unlabeled data can be used to generate better data representation that helps improve DDL's phenotyping performance over existing methods that only uses labeled data.
Tianfan Fu, Trong Nghia Hoang, Cao Xiao, Jimeng Sun 0001
IJCAI3
2019 RDPD: Rich Data Helps Poor Data via Imitation
abstract
In many situations, we need to build and deploy separate models in related environments with different data qualities. For example, an environment with strong observation equipments (e.g., intensive care units) often provides high-quality multi-modal data, which are acquired from multiple sensory devices and have rich-feature representations. On the other hand, an environment with poor observation equipment (e.g., at home) only provides low-quality, uni-modal data with poor-feature representations. To deploy a competitive model in a poor-data environment without requiring direct access to multi-modal data acquired from a rich-data environment, this paper develops and presents a knowledge distillation (KD) method (RDPD) to enhance a predictive model trained on poor data using knowledge distilled from a high-complexity model trained on rich, private data. We evaluated RDPD on three real-world datasets and shown that its distilled model consistently outperformed all baselines across all datasets, especially achieving the greatest performance improvement over a model trained only on low-quality data by 24.56% on PR-AUC and 12.21% on ROC-AUC, and over that of a state-of-the-art KD model by 5.91% on PR-AUC and 4.44% on ROC-AUC.
Shenda Hong, Cao Xiao, Trong Nghia Hoang, Tengfei Ma 0001, Hongyan Li 0002, Jimeng Sun 0001
IJCAI2
2019 MINA: Multilevel Knowledge-Guided Attention for Modeling Electrocardiography Signals
abstract
Electrocardiography (ECG) signals are commonly used to diagnose various cardiac abnormalities. Recently, deep learning models showed initial success on modeling ECG data, however they are mostly black-box, thus lack interpretability needed for clinical usage. In this work, we propose MultIlevel kNowledge-guided Attention networks (MINA) that predict heart diseases from ECG signals with intuitive explanation aligned with medical knowledge. By extracting multilevel (beat-, rhythm- and frequency-level) domain knowledge features separately, MINA combines the medical knowledge and ECG data via a multilevel attention model, making the learned models highly interpretable. Our experiments showed MINA achieved PR-AUC 0.9436 (outperforming the best baseline by 5.51%) in real world ECG dataset. Finally, MINA also demonstrated robust performance and strong interpretability against signal distortion and noise contamination.
Shenda Hong, Cao Xiao, Tengfei Ma 0001, Hongyan Li 0002, Jimeng Sun 0001
IJCAI2
2019 Pre-training of Graph Augmented Transformers for Medication Recommendation
abstract
Medication recommendation is an important healthcare application. It is commonly formulated as a temporal prediction task. Hence, most existing works only utilize longitudinal electronic health records (EHRs) from a small number of patients with multiple visits ignoring a large number of patients with a single visit (selection bias). Moreover, important hierarchical knowledge such as diagnosis hierarchy is not leveraged in the representation learning process. Despite the success of deep learning techniques in computational phenotyping, most previous approaches have two limitations: task-oriented representation and ignoring hierarchies of medical codes. To address these challenges, we propose G-BERT, a new model to combine the power of Graph Neural Networks (GNNs) and BERT (Bidirectional Encoder Representations from Transformers) for medical code representation and medication recommendation. We use GNNs to represent the internal hierarchical structures of medical codes. Then we integrate the GNN representation into a transformer-based visit encoder and pre-train it on EHR data from patients only with a single visit. The pre-trained visit encoder and representation are then fine-tuned for downstream predictive tasks on longitudinal EHRs from patients with multiple visits. G-BERT is the first to bring the language model pre-training schema into the healthcare domain and it achieved state-of-the-art performance on the medication recommendation task.
Junyuan Shang, Tengfei Ma 0001, Cao Xiao, Jimeng Sun 0001
IJCAI3
2019 Retaining Privileged Information for Multi-Task Learning
abstract
Knowledge transfer has been of great interest in current machine learning research, as many have speculated its importance in modeling the human ability to rapidly generalize learned models to new scenarios. Particularly in cases where training samples are limited, knowledge transfer shows improvement on both the learning speed and generalization performance of related tasks. Recently, Learning Using Privileged Information (LUPI) has presented a new direction in knowledge transfer by modeling the transfer of prior knowledge as a Teacher-Student interaction process. Under LUPI, a Teacher model uses Privileged Information (PI) that is only available at training time to improve the sample complexity required to train a Student learner for a given task. In this work, we present a LUPI formulation that allows privileged information to be retained in a multi-task learning setting. We propose a novel feature matching algorithm that projects samples from the original feature space and the privilege information space into a joint latent space in a way that informs similarity between training samples. Our experiments show that useful knowledge from PI is maintained in the latent space and greatly improves the sample efficiency of other related learning tasks. We also provide an analysis of sample complexity of the proposed LUPI method, which under some favorable assumptions can achieve a greater sample efficiency than brute force methods.
Fengyi Tang, Cao Xiao, Fei Wang 0001, Li-Wei H. Lehman
KDD2
2019 Tutorial: Data Mining Methods for Drug Discovery and Development
abstract
In silico modeling of medicine refers to the direct use of computational methods in support of drug discovery and development. Machine learning and data mining methods have become an integral part of in silico modeling and demonstrated promising performance at various phases of the drug discovery and development process. In this tutorial we will introduce data analytic methods in drug discovery and development. For the first half, we will provide an overview about related data and analytic tasks, and then present the enabling data analytic methods for these tasks. For the second half, we will describe concrete applications of each of those tasks. The tutorial will be concluded with open problems and a Q&A session.
Cao Xiao, Jimeng Sun 0001
KDD1
2019 Longitudinal Adversarial Attack on Electronic Health Records Data
abstract
Although deep learning models trained on electronic health records (EHR) data have shown state-of-the-art performance in many predictive clinical tasks, the discovery of adversarial examples (i.e., input data that are engineered to cause misclassification) has exposed vulnerabilities with lab and imaging data. We specifically consider adversarial examples with longitudinal EHR data, an area that has not been previously examined because of the challenges with temporal high-dimensional and sparse features. We propose Longitudinal AdVersarial Attack (, a saliency score based adversarial example using a method that requires a minimal number of perturbations and that automatically minimizes the likelihood of detection. Features are selected and modified by jointly modeling a saliency map and attention mechanism. Experimental results with longitudinal EHR data show that an substantially reduce model performance for attention-based target models (from AUPR = 0.5 to AUPR = 0.08).
Sungtae An, Cao Xiao, Walter F. Stewart, Jimeng Sun 0001
WWW2
2018 Heterogeneous Hyper-Network Embedding
abstract
Heterogeneous hyper-networks is used to represent multi-modal and composite interactions between data points. In such networks, several different types of nodes form a hyperedge. Heterogeneous hyper-network embedding learns a distributed node representation under such complex interactions while preserving the network structure. However, this is a challenging task due to the multiple modalities and composite interactions. In this study, a deep approach is proposed to embed heterogeneous attributed hyper-networks with complicated and non-linear node relationships. In particular, a fully-connected and graph convolutional layers are designed to project different types of nodes into a common low-dimensional space, a tuple-wise similarity function is proposed to preserve the network structure, and a ranking based loss function is used to improve the similarity scores of hyperedges in the embedding space. The proposed approach is evaluated on synthetic and real world datasets and a better performance is obtained compared with baselines.
Inci M. Baytas, Cao Xiao, Fei Wang 0001, Anil K. Jain 0001
ICDM2
2018 FastGCN: Fast Learning with Graph Convolutional Networks via Importance Sampling
Jie Chen 0007, Tengfei Ma 0001, Cao Xiao
ICLR (Poster)3
2018 Drug Similarity Integration Through Attentive Multi-view Graph Auto-Encoders
abstract
Drug similarity has been studied to support downstream clinical tasks such as inferring novel properties of drugs (e.g. side effects, indications, interactions) from known properties. The growing availability of new types of drug features brings the opportunity of learning a more comprehensive and accurate drug similarity that represents the full spectrum of underlying drug relations. However, it is challenging to integrate these heterogeneous, noisy, nonlinear-related information to learn accurate similarity measures especially when labels are scarce. Moreover, there is a trade-off between accuracy and interpretability. In this paper, we propose to learn accurate and interpretable similarity measures from multiple types of drug features. In particular, we model the integration using multi-view graph auto-encoders, and add attentive mechanism to determine the weights for each view with respect to corresponding tasks and features for better interpretability. Our model has flexible design for both semi-supervised and unsupervised settings. Experimental results demonstrated significant predictive accuracy improvement. Case studies also showed better model capacity (e.g. embed node features) and interpretability.
Tengfei Ma 0001, Cao Xiao, Fei Wang 0001
IJCAI2
2018 Pairwise-Ranking based Collaborative Recurrent Neural Networks for Clinical Event Prediction
abstract
Patient Electronic Health Records (EHR) data consist of sequences of patient visits over time. Sequential prediction of patients' future clinical events (e.g., diagnoses) from their historical EHR data is a core research task and motives a series of predictive models including deep learning. The existing research mainly adopts a classification framework, which treats the observed and unobserved events as positive and negative classes. However, this may not be true in real clinical setting considering the high rate of missed diagnoses and human errors. In this paper, we propose to formulate the clinical event prediction problem as an events recommendation problem. An end-to-end pairwise-ranking based collaborative recurrent neural networks (PacRNN) is proposed to solve it, which firstly embeds patient clinical contexts with attention RNN, then uses Bayesian Personalized Ranking (BPR) regularized by disease co-occurrence to rank probabilities of patient-specific diseases, as well as use point process to provide simultaneous prediction of the occurring time of these diagnoses. Experimental results on two real world EHR datasets demonstrate the robust performance, interpretability, and efficacy of PacRNN.
Zhi Qiao 0007, Shiwan Zhao, Cao Xiao, Xiang Li 0013, Fei Wang 0001
IJCAI3
2018 MiME: Multilevel Medical Embedding of Electronic Health Records for Predictive Healthcare
abstract
Deep learning models exhibit state-of-the-art performance for many predictive healthcare tasks using electronic health records (EHR) data, but these models typically require training data volume that exceeds the capacity of most healthcare systems. External resources such as medical ontologies are used to bridge the data volume constraint, but this approach is often not directly applicable or useful because of inconsistencies with terminology. To solve the data insufficiency challenge, we leverage the inherent multilevel structure of EHR data and, in particular, the encoded relationships among medical codes. We propose Multilevel Medical Embedding (MiME) which learns the multilevel embedding of EHR data while jointly performing auxiliary prediction tasks that rely on this inherent EHR structure without the need for external labels. We conducted two prediction tasks, heart failure prediction and sequential disease prediction, where MiME outperformed baseline methods in diverse evaluation settings. In particular, MiME consistently outperformed all baselines when predicting heart failure on datasets of different volumes, especially demonstrating the greatest performance improvement (15% relative gain in PR-AUC over the best baseline) on the smallest dataset, demonstrating its ability to effectively model the multilevel structure of EHR data.
Edward Choi 0003, Cao Xiao, Walter F. Stewart, Jimeng Sun 0001
NeurIPS2
2018 Constrained Generation of Semantically Valid Graphs via Regularizing Variational Autoencoders
abstract
Deep generative models have achieved remarkable success in various data domains, including images, time series, and natural languages. There remain, however, substantial challenges for combinatorial structures, including graphs. One of the key challenges lies in the difficulty of ensuring semantic validity in context. For example, in molecular graphs, the number of bonding-electron pairs must not exceed the valence of an atom; whereas in protein interaction networks, two proteins may be connected only when they belong to the same or correlated gene ontology terms. These constraints are not easy to be incorporated into a generative model. In this work, we propose a regularization framework for variational autoencoders as a step toward semantic validity. We focus on the matrix representation of graphs and formulate penalty terms that regularize the output distribution of the decoder to encourage the satisfaction of validity constraints. Experimental results confirm a much higher likelihood of sampling valid graphs in our approach, compared with others reported in the literature.
Tengfei Ma 0001, Jie Chen 0007, Cao Xiao
NeurIPS3
2018 Health-ATM: A Deep Architecture for Multifaceted Patient Health Record Representation and Risk Prediction
abstract
Leveraging massive electronic health records (EHR) brings tremendous promises to advance clinical and precision medicine informatics research. However, it is very challenging to directly work with multifaceted patient information encoded in their EHR data. Deriving effective representations of patient EHRs is a crucial step to bridge raw EHR information and the endpoint analytical tasks, such as risk prediction or disease subtyping. In this paper, we propose Health-ATM, a novel and integrated deep architecture to uncover patients' comprehensive health information from their noisy, longitudinal, heterogeneous and irregular EHR data. Health-ATM extracts comprehensive multifaceted patient information patterns with attentive and time-aware modulars (ATM) and a hybrid network structure composed of both Recurrent Neural Network (RNN) and Convolutional Neural Network (CNN). The learned features are finally fed into a prediction layer to conduct the risk prediction task. We evaluated the Health-ATM on both artificial and real world EHR corpus and demonstrated its promising utility and efficacy on representation learning and disease onset predictions.
Tengfei Ma 0001, Cao Xiao, Fei Wang 0001
SDM2
2018 Opportunities and challenges in developing deep learning models using electronic health records data: a systematic review
abstract
Objective: To conduct a systematic review of deep learning models for electronic health record (EHR) data, and illustrate various deep learning architectures for analyzing different data sources and their target applications. We also highlight ongoing research and identify open challenges in building deep learning models of EHRs. Design/method: We searched PubMed and Google Scholar for papers on deep learning studies using EHR data published between January 1, 2010, and January 31, 2018. We summarize them according to these axes: types of analytics tasks, types of deep learning model architectures, special challenges arising from health data and tasks and their potential solutions, as well as evaluation strategies. Results: We surveyed and analyzed multiple aspects of the 98 articles we found and identified the following analytics tasks: disease detection/classification, sequential prediction of clinical events, concept embedding, data augmentation, and EHR data privacy. We then studied how deep architectures were applied to these tasks. We also discussed some special challenges arising from modeling EHR data and reviewed a few popular approaches. Finally, we summarized how performance evaluations were conducted for each task. Discussion: Despite the early success in using deep learning for health analytics applications, there still exist a number of issues to be addressed. We discuss them in detail including data and label availability, the interpretability and transparency of the model, and ease of deployment.
Cao Xiao, Edward Choi 0003, Jimeng Sun 0001
J. Am. Medical Informatics Assoc.1
2018 Optimal Expert Knowledge Elicitation for Bayesian Network Structure Identification
abstract
Bayesian network (BN) has been a popular tool for gaining mechanistic understanding of variables by revealing how the variables influence each other. It has been found very effective in a few studies in quality control and process monitoring. However, for complex problems where the structure of a BN is unknown, a common approach is to learn the BN structure from observational data. A fundamental bottleneck of this approach is that observational data can only be used to discover part of the influential relationships among variables. To overcome this problem, we propose to combine observational data and expert knowledge. To the best of the author's knowledge, our approach is the first of its kind that formulates an experimental design framework to automate the expert elicitation process and collect the most informative expert knowledge, optimally matched to the observational data, to learn the BN structure.
Cao Xiao, Ji Liu 0002, Bo Zeng 0001, Shuai Huang 0001
IEEE Trans Autom. Sci. Eng.1
2017 Multitask Dyadic Prediction and Its Application in Prediction of Adverse Drug-Drug Interaction
abstract
Adverse drug-drug interactions (DDIs) remain a leading cause of morbidity and mortality around the world. Identifying potential DDIs during the drug design process is critical in guiding targeted clinical drug safety testing. Although detection of adverse DDIs is conducted during Phase IV clinical trials, there are still a large number of new DDIs founded by accidents after the drugs were put on market. With the arrival of big data era, more and more pharmaceutical research and development data are becoming available, which provides an invaluable resource for digging insights that can potentially be leveraged in early prediction of DDIs. Many computational approaches have been proposed in recent years for DDI prediction. However, most of them focused on binary prediction (with or without DDI), despite the fact that each DDI is associated with a different type. Predicting the actual DDI type will help us better understand the DDI mechanism and identify proper ways to prevent it. In this paper, we formulate the DDI type prediction problem as a multitask dyadic regression problem, where the prediction of each specific DDI type is treated as a task. Compared with conventional matrix completion approaches which can only impute the missing entries in the DDI matrix, our approach can directly regress those dyadic relationships (DDIs) and thus can be extend to new drugs more easily. We developed an effective proximal gradient method to solve the problem. Evaluation on real world datasets is presented to demonstrate the effectiveness of the proposed approach.
Bo Jin 0001, Cao Xiao, Ping Zhang 0016, Xiaopeng Wei, Fei Wang 0001
AAAI3
2017 Adverse Drug Reaction Prediction with Symbolic Latent Dirichlet Allocation
abstract
Adverse drug reaction (ADR) is a major burden for patients and healthcare industry. It usually causes preventable hospitalizations and deaths, while associated with a huge amount of cost. Traditional preclinical in vitro safety profiling and clinical safety trials are restricted in terms of small scale, long duration, huge financial costs and limited statistical signifi- cance. The availability of large amounts of drug and ADR data potentially allows ADR predictions during the drugs’ early preclinical stage with data analytics methods to inform more targeted clinical safety tests. Despite their initial success, existing methods have trade-offs among interpretability, predictive power and efficiency. This urges us to explore methods that could have all these strengths and provide practical solutions for real world ADR predictions. We cast the ADR-drug relation structure into a three-layer hierarchical Bayesian model. We interpret each ADR as a symbolic word and apply latent Dirichlet allocation (LDA) to learn topics that may represent certain biochemical mechanism that relates ADRs with drug structures. Based on LDA, we designed an equivalent regularization term to incorporate the hierarchical ADR domain knowledge. Finally, we developed a mixed input model leveraging a fast collapsed Gibbs sampling method that the complexity of each iteration of Gibbs sampling proportional only to the number of positive ADRs. Experiments on real world data show our models achieved higher prediction accuracy and shorter running time than the state-of-the-art alternatives.
Cao Xiao, Ping Zhang 0016, W. Art Chaovalitwongse, Jianying Hu, Fei Wang 0001
AAAI1
2017 Subtyping Parkinson's Disease with Recurrent Neural Network Models
Jian Liang 0001, Cao Xiao, Yize Zhao, Fei Wang 0001
AMIA3
2017 Patient Subtyping via Time-Aware LSTM Networks
abstract
In the study of various diseases, heterogeneity among patients usually leads to different progression patterns and may require different types of therapeutic intervention. Therefore, it is important to study patient subtyping, which is grouping of patients into disease characterizing subtypes. Subtyping from complex patient data is challenging because of the information heterogeneity and temporal dynamics. Long-Short Term Memory (LSTM) has been successfully used in many domains for processing sequential data, and recently applied for analyzing longitudinal patient records. The LSTM units are designed to handle data with constant elapsed times between consecutive elements of a sequence. Given that time lapse between successive elements in patient records can vary from days to months, the design of traditional LSTM may lead to suboptimal performance. In this paper, we propose a novel LSTM unit called Time-Aware LSTM (T-LSTM) to handle irregular time intervals in longitudinal patient records. We learn a subspace decomposition of the cell memory which enables time decay to discount the memory content according to the elapsed time. We propose a patient subtyping model that leverages the proposed T-LSTM in an auto-encoder to learn a powerful single representation for sequential records of patients, which are then used to cluster patients into clinical subtypes. Experiments on synthetic and real world datasets show that the proposed T-LSTM architecture captures the underlying structures in the sequences with time irregularities.
Inci M. Baytas, Cao Xiao, Fei Wang 0001, Anil K. Jain 0001
KDD2
2017 An RNN Architecture with Dynamic Temporal Matching for Personalized Predictions of Parkinson's Disease
abstract
Parkinson's disease (PD) is a chronic disease that develops over years and varies dramatically in its clinical manifestations. A preferred strategy to resolve this heterogeneity and thus enable better prognosis and targeted therapies is to segment out more homogeneous patient sub-populations. However, it is challenging to evaluate the clinical similarities among patients because of the longitudinality and temporality of their records. To address this issue, we propose a deep model that directly learns patient similarity from longitudinal and multi-modal patient records with an Recurrent Neural Network (RNN) architecture, which learns the similarity between two longitudinal patient record sequences through dynamically matching temporal patterns in patient sequences. Evaluations on real world patient records demonstrate the promising utility and efficacy of the proposed architecture in personalized predictions.
Chao Che, Cao Xiao, Jian Liang 0002, Bo Jin 0001, Jiayu Zho, Fei Wang 0001
SDM2
2017 Unsupervised Sequential Outlier Detection With Deep Architectures
abstract
Unsupervised outlier detection is a vital task and has high impact on a wide variety of applications domains, such as image analysis and video surveillance. It also gains long-standing attentions and has been extensively studied in multiple research areas. Detecting and taking action on outliers as quickly as possible are imperative in order to protect network and related stakeholders or to maintain the reliability of critical systems. However, outlier detection is difficult due to the one class nature and challenges in feature construction. Sequential anomaly detection is even harder with more challenges from temporal correlation in data, as well as the presence of noise and high dimensionality. In this paper, we introduce a novel deep structured framework to solve the challenging sequential outlier detection problem. We use autoencoder models to capture the intrinsic difference between outliers and normal instances and integrate the models to recurrent neural networks that allow the learning to make use of previous context as well as make the learners more robust to warp along the time axis. Furthermore, we propose to use a layerwise training procedure, which significantly simplifies the training procedure and hence helps achieve efficient and scalable training. In addition, we investigate a fine-tuning step to update all parameters set by incorporating the temporal correlation in the sequence. We further apply our proposed models to conduct systematic experiments on five real-world benchmark data sets. Experimental results demonstrate the effectiveness of our model, compared with other state-of-the-art approaches.
Weining Lu, Yu Cheng 0001, Cao Xiao, Shiyu Chang, Shuai Huang 0001, Bin Liang 0001, Thomas S. Huang
IEEE Trans. Image Process.3
2016 An Efficient Time Series Subsequence Pattern Mining and Prediction Framework with an Application to Respiratory Motion Prediction
abstract
Traditional time series analysis methods are limited on some complex real-world time series data. Respiratory motion prediction is one of such challenging problems. The memory-based nearest neighbor approaches haveshown potentials in predicting complex nonlinear time series compared to many traditional parametric prediction models. However, the massive time series subsequences representation, the similarity distance measures, the number of nearest neighbors, and the ensemble functions create challenges as well as limit the performance of nearest neighbor approaches in complex time series prediction. To address these problems, we propose a flexible time series pattern representation and selection framework, called the orthogonalpolynomial-based variant-nearest-neighbor (OPVNN) approach. For the respiratory motion prediction problem, the proposed approach achieved the highest and most robust prediction performance compared to the state-of-the-art time series prediction methods. With a solid mathematical and theoretical foundation in orthogonal polynomials, the proposed time series representation, subsequence pattern mining and prediction framework has a great potential to benefit those industry and medical applications that need to handle highly nonlinear and complex time series data streams, such as quasi-periodic ones.
KinMing Kam, Cao Xiao, Stephen R. Bowen, W. Art Chaovalitwongse
AAAI3
2016 A Patient-Specific Model for Predicting Tibia Soft Tissue Insertions From Bony Outlines Using a Spatial Structure Supervised Learning Framework
abstract
Recreating the natural anatomy in ligament reconstruction is crucial to fully restore the knee joint function and reduce impingement on iatrogenic injury to adjacent structures, yet is subject to the difficulties in locating ligament and other associated soft tissues insertion sites intraoperatively and the high interperson morphological variability cross patients. In this study, we present a new quantitative analysis method capable of achieving personalized identification of cruciate ligament and soft tissue insertions. We craft patient-specific features of tibia outline that can be accurately and reliably measured from CT images. In addition, we propose a supervised structure learning and prediction model with special interdimensional and response structure regularization terms to capture relationship between the spatial arrangement of soft tissue insertions and the patient-specific features extracted from the tibia outlines. In the experiment, the proposed model outperforms baseline models and provides an accurate and accessible approach that can be used as the first and the most critical step to achieve personalized surgical planning in cruciate ligament reconstruction.
Cao Xiao, Liying Zheng, W. Art Chaovalitwongse
IEEE Trans. Hum. Mach. Syst.1
2016 Optimization Models for Feature Selection of Decomposed Nearest Neighbor
abstract
The traditional k-nearest neighbor (KNN) method classifies an object by the majority vote of its neighbors, and only one parameter k is to be optimized. We propose a decomposed version of the KKN (DKNN) method, which classifies an unseen object based on the distances to the centroids of the KNNs. DKNN has a training process to learn the local-optimafree distance metric by solving a convex optimization problem. The optimization problem not only learns a metric that minimizes classification errors and maximizes the margin between intraclass and interclass distances. In addition, it also selects important features and removes irrelevant features when L1 regularization is incorporated in the optimization model. The feature selection component is extremely useful for high-dimensional classification problems. Tested on public datasets, the proposed DKNN is competitive and often outperforms traditional machine learning algorithms.
Cao Xiao, W. Art Chaovalitwongse
IEEE Trans. Syst. Man Cybern. Syst.1