Personalized Dictionary Learning for Heterogeneous Datasets
Geyu Liang, Naichen Shi, Raed Al Kontar, Salar Fattahi
Abstract
We introduce a relevant yet challenging problem named Personalized Dictionary Learning (PerDL), where the goal is to learn sparse linear representations from heterogeneous datasets that share some commonality. In PerDL, we model each dataset's shared and unique features as global and local dictionaries. Challenges for PerDL not only are inherited from classical dictionary learning (DL), but also arise due to the unknown nature of the shared and unique features. In this paper, we rigorously formulate this problem and provide conditions under which the global and local dictionaries can be provably disentangled. Under these conditions, we provide a meta-algorithm called Personalized Matching and Averaging (PerMA) that can recover both global and local dictionaries from heterogeneous datasets. PerMA is highly efficient; it converges to the ground truth at a linear rate under suitable conditions. Moreover, it automatically borrows strength from strong learners to improve the prediction of weak learners. As a general framework for extracting global and local dictionaries, we show the application of PerDL in different learning tasks, such as training with imbalanced datasets and video surveillance.
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.
Cited by top-tier papers1
Ask how each one uses itBuilds on4
- SCAFFOLD: Stochastic Controlled Averaging for Federated LearningSai Praneeth Karimireddy, Satyen Kale, Mehryar Mohri, Sashank J. Reddi et al.ICML 2020 · 3,875 citations
- Personalized Federated Learning with Moreau EnvelopesCanh T. Dinh, Nguyen Hoang Tran, Tuan Dung NguyenNeurIPS 2020 · 1,542 citations
- Federated Learning with Matched AveragingHongyi Wang, Mikhail Yurochkin, Yuekai Sun, Dimitris S. Papailiopoulos et al.ICLR 2020 · 1,368 citations
- Ditto: Fair and Robust Federated Learning Through PersonalizationTian Li, Shengyuan Hu, Ahmad Beirami, Virginia SmithICML 2021 · 1,313 citations
Related papers
- DePRL: Achieving Linear Convergence Speedup in Personalized Decentralized Learning with Shared RepresentationsGuojun Xiong, Gang Yan, Shiqiang Wang, Jian LiAAAI 2024 · 9 citations
- The Best of Both Worlds: Accurate Global and Personalized Models through Federated Learning with Data-Free Hyper-Knowledge DistillationHuancheng Chen, Chianing Wang, Haris VikaloICLR 2023 · 11 citations
- Personalized Federated Learning via Feature Distribution AdaptationConnor Mclaughlin, Lili SuNeurIPS 2024 · 44 citations
- Unbalanced Multi-view Deep LearningCai Xu, Zehui Li, Ziyu Guan, Wei Zhao et al.ACM MM 2023 · 5 citations
- FedFSL-CFRD: Personalized Federated Few-Shot Learning with Collaborative Feature Representation DisentanglementShanfeng Wang, Jianzhao Li, Zaitian Liu, Yourun Zhang et al.AAAI 2025 · 1 citation
