CMVAE: Contrastive Supervision Improves Single-Omic Latent Manifold for Breast Cancer Survival Prognosis
Akash SinghiD, Adrian Kohut, John Michael Templeton, Anowarul KabiriD
Abstract
High-dimensional omics data are vital for fore- casting patient prognosis in oncology. While deep multi-omics learning methods are increasingly popular, they suffer from data missingness and low patient overlap across modalities. Consequently, advancing single-omic representation learning re- mains a critical yet challenging research problem. Standard generative dimensionality reduction methods rely on recon- struction objectives that optimize the global manifold topology; however, overlook downstream clinical outcomes. To address these limitations, we introduce a Contrastive Masked Denois- ing Variational Autoencoder (CMVAE), a self-supervised dual- optimization framework designed to map single-omic information into a clinical outcome-aware latent space. A stochastic feature masking strategy corrupts the input omic profiles, and CMVAE is optimized to simultaneously infer missing molecular data and contrastive alignment of patient survival trajectories. We evaluate CMVAE using leak-proof five-fold stratified cross-validation on the TCGA Breast Invasive Carcinoma (BRCA) cohort. The method outperforms traditional unsupervised methods, achieving an AUROC of 0.7397 {+/-} 0.0239 on patient prognosis. Further analysis demonstrates a highly significant (p < 0.0001) risk stratification among low vs. high risk groups. Beyond comparing within single-omic methods, CMVAE outperforms six state-of- the-art multi-omics survival baselines by a minimum of 3%. Availability: All codes including data preparation scripts, training and validation recipes, and experimental configurations are available at: https://github.com/kabir-ai2bio-lab/CMVAE. Note: Accepted at the IEEE BIBM 2026 workshop RLPD-Bio (Representation Learning and Pattern Discovery for Bioinformatics and Biomedicine).
02 /Specification & Classification
Click any tag to browse matching papers- Organism
- Assay
- Research Focus
- Approach
Reproducibility ledger
Nobody has reported trying to reproduce this paper yet. Did you replicate an experiment or run the code? Your report saves the next lab a week.
Sign in to file a reproducibility report.
Related papers
- Integrating multi-omics data through deep learning for accurate cancer prognosis prediction
Hua Chai, Xiang Zhou, … Yuedong Yang·Computers in Biology and MedicinePeer-reviewed·May 9, 2021
- Deep representation learning for temporal inference in cancer omics: a systematic review
Guillermo Prol-Castelo, Davide Cirillo, Alfonso Valencia·bioRxiv·May 29, 2025
- Breast Cancer Subtyping with HyperCLSA: A Hypergraph Contrastive Learning Pipeline for Multi-Omics Data Integration
Gaurav Bhole, Poorvi HC, … Prabhakar Bhimalapuram·Lecture notes in computer sciencePeer-reviewed·Jan 1, 2026
- Deep learning based on multi-omics integration identifies potential therapeutic targets in breast cancer
Xingxin Pan, Brandon Burgman, … S. Stephen Yi·bioRxiv·Jan 18, 2022
- MOTLAB: A Weighted Multi-Omics Transfer Learning Approach to Mitigate Breast Cancer Racial Disparities
Min-Jeong Baek, Lusheng Li, … Shibiao Wan·bioRxiv·Sep 24, 2026
- Hierarchical autoencoder-based integration improves performance in multi-omics cancer survival models through soft modality selection
David Wissel, Daniel Rowson, Valentina Boeva·bioRxiv·Sep 16, 2021