Back

A Generalized Supervised Contrastive Learning Framework for Integrative Multi-omics Prediction Models

Yang, S.; Wang, S.; Wang, Y.; Rong, R.; Li, B.; Koh, A. I.; Xiao, G.; Liu, D.; Zhan, X.

2023-11-04 bioinformatics
10.1101/2023.11.01.565241 bioRxiv
Show abstract

Recent technological advances have highlighted the significant impact of the human microbiome and metabolites on physiological conditions. Integrating microbiome and metabolite data has shown promise in predictive capabilities. We developed a new supervised contrastive learning framework, MB-SupCon-cont, that (1) proposes a general contrastive learning framework for continuous outcomes and (2) improves prediction accuracy over models using single omics data. Simulation studies confirmed the improved performance of MB-SupCon-cont, and applied scenarios in type 2 diabetes and high-fat diet studies also showed improved prediction performance. Overall, MB-SupCon-cont is a versatile research tool for multi-omics prediction models.

Published in Frontiers in Microbiomes · not in our set (fewer than 10 published preprints to learn from) · training set

Matching journals

The top 3 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.