Optimal Transfer Learning for Missing Not-at-Random Matrix Completion
Akhil Jalan, Yassir Jedra, Arya Mazumdar, Soumendu Sundar Mukherjee, Purnamrita Sarkar
Abstract
We study transfer learning for matrix completion in a Missing Not-at-Random (MNAR) setting that is motivated by biological problems. The target matrix Q has entire rows and columns missing, making estimation impossible without side information. To address this, we use a noisy and incomplete source matrix P , which relates to Q via a feature shift in latent space. We consider both the active and passive sampling of rows and columns. We establish minimax lower bounds for entrywise estimation error in each setting. Our computationally efficient estimation framework achieves this lower bound for the active setting, which leverages the source data to query the most informative rows and columns of Q. This avoids the need for incoherence assumptions required for rate optimality in the passive sampling setting. We demonstrate the effectiveness of our approach through comparisons with existing algorithms on real-world biological datasets.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 0e2fb848-4e82-424c-a633-fecafaa4366dCited by top-tier papers1
Ask how each one uses itBuilds on7
- What is being transferred in transfer learning?Behnam Neyshabur, Hanie Sedghi, Chiyuan ZhangNeurIPS 2020 · 654 citations
- On the Theory of Transfer Learning: The Importance of Task DiversityNilesh Tripuraneni, Michael I. Jordan, Chi JinNeurIPS 2020 · 263 citations
- Learning with Good Feature Representations in Bandits and in RL with a Generative ModelTor Lattimore, Csaba Szepesvári, Gellért WeiszICML 2020 · 181 citations
- Active Learning for Domain Adaptation: An Energy-Based ApproachBinhui Xie, Longhui Yuan, Shuang Li, Chi Harold Liu et al.AAAI 2022 · 149 citations
- not-MIWAE: Deep Generative Modelling with Missing not at Random DataNiels Bruun Ipsen, Pierre-Alexandre Mattei, Jes FrellsenICLR 2021 · 81 citations
Related papers
- Transfer Learning for Latent Variable Network ModelsAkhil Jalan, Arya Mazumdar, Soumendu Sundar Mukherjee, Purnamrita SarkarNeurIPS 2024 · 5 citations
- Matrix Completion with Incomplete Side Information via Orthogonal Complement ProjectionGengshuo Chang, Wei Zhang, Lehan ZhangICML 2025
- Causal Matrix Completion under Multiple Treatments via Mixed Synthetic Nearest NeighborsMinrui Luo, Zhiheng ZhangICML 2026
- Matrix Completion with Model-free WeightingJiayi Wang, Raymond K. W. Wong, Xiaojun Mao, Kwun Chuen Gary ChanICML 2021 · 7 citations
- Fine-grained Generalization Analysis of Inductive Matrix CompletionAntoine Ledent, Rodrigo Alves, Yunwen Lei, Marius KloftNeurIPS 2021 · 14 citations
