Multiple Augmented Reduced Rank Regression for Pan-Cancer Analysis
Statistical approaches that successfully combine multiple datasets are more powerful, efficient, and scientifically informative than separate analyses. To address variation architectures correctly and comprehensively for high-dimensional data across multiple sample sets (i.e., cohorts), we propose multiple augmented reduced rank regression (maRRR), a flexible matrix regression and factorization method to concurrently learn both covariate-driven and auxiliary structured variation. We consider a structured nuclear norm objective that is motivated by random matrix theory, in which the regression or factorization terms may be shared or specific to any number of cohorts. Our framework subsumes several existing methods, such as reduced rank regression and unsupervised multi-matrix factorization approaches, and includes a promising novel approach to regression and factorization of a single dataset (aRRR) as a special case. Simulations demonstrate substantial gains in power from combining multiple datasets, and from parsimoniously accounting for all structured variation. We apply maRRR to gene expression data from multiple cancer types (i.e., pan-cancer) from TCGA, with somatic mutations as covariates. The method performs well with respect to prediction and imputation of held-out data, and provides new insights into mutation-driven and auxiliary variation that is shared or specific to certain cancer types.
Code (1)
Tasks
ImputationregressionSimilar Papers 제목 키워드 기반
LoRANN: Low-Rank Matrix Factorization for Approximate Nearest Neighbor Search
Approximate nearest neighbor (ANN) search is a key component in many modern machine learning pipelines; recent use cases include retrieval-augmented generation (RAG) and vector databases. Clustering-based ANN algorithms,…
ClusteringGPUQuantizationRAG+2Simultaneous Best Subset Selection and Dimension Reduction via Primal-Dual Iterations
Sparse reduced rank regression is an essential statistical learning method. In the contemporary literature, estimation is typically formulated as a nonconvex optimization that often yields to a local optimum in numerical…
Dimensionality ReductionregressionFrom Gene Expression to Drug Response: A Collaborative Filtering Approach
Predicting the response of cancer cells to drugs is an important problem in pharmacogenomics. Recent efforts in generation of large scale datasets profiling gene expression and drug sensitivity in cell lines have provide…
Collaborative FilteringSensitivityVector-Valued Least-Squares Regression under Output Regularity Assumptions
We propose and analyse a reduced-rank method for solving least-squares regression problems with infinite dimensional output. We derive learning bounds for our method, and study under which setting statistical performance…
Image ReconstructionMulti-Label ClassificationMUlTI-LABEL-ClASSIFICATIONregression+1A Generative AI Approach for Reducing Skin Tone Bias in Skin Cancer Classification
Skin cancer is one of the most common cancers worldwide and early detection is critical for effective treatment. However, current AI diagnostic tools are often trained on datasets dominated by lighter skin tones, leading…
Skin Cancer ClassificationBinary ClassificationLesion SegmentationData Augmentation