Timezone: »

Learning Task-Agnostic Embedding of Multiple Black-Box Experts for Multi-Task Model Fusion
Nghia Hoang · Thanh Lam · Bryan Kian Hsiang Low · Patrick Jaillet

Thu Jul 16 07:00 AM -- 07:45 AM & Thu Jul 16 07:00 PM -- 07:45 PM (PDT) @ Virtual #None

Model fusion is an emerging study in collective learning where heterogeneous experts with private data and learning architectures need to combine their black-box knowledge for better performance. Existing literature achieves this via a local knowledge distillation scheme that transfuses the predictive patterns of each pre-trained expert onto a white-box imitator model, which can be incorporated efficiently into a global model. This scheme however does not extend to multi-task scenarios where different experts were trained to solve different tasks and only part of their distilled knowledge is relevant to a new task. To address this multi-task challenge, we develop a new fusion paradigm that represents each expert as a distribution over a spectrum of predictive prototypes, which are isolated from task-specific information encoded within the prototype distribution. The task-agnostic prototypes can then be reintegrated to generate a new model that solves a new task encoded with a different prototype distribution. The fusion and adaptation performance of the proposed framework is demonstrated empirically on several real-world benchmark datasets.

Author Information

Nghia Hoang (MIT-IBM Watson AI Lab, IBM Research)
Thanh Lam (National University of Singapore)
Bryan Kian Hsiang Low (National University of Singapore)

Dr. Bryan Low is an Associate Professor of Computer Science at the National University of Singapore and the Deputy Director of AI Research at AI Singapore. He obtained the B.Sc. (Hons.) and M.Sc. degrees in Computer Science from National University of Singapore, Singapore, in 2001 and 2002, respectively, and the Ph.D. degree in Electrical and Computer Engineering from Carnegie Mellon University, Pittsburgh, Pennsylvania, in 2009. His research interests include probabilistic & automated machine learning, planning under uncertainty, and multi-agent/robot systems. Dr. Low is the recipient of the (1) Andrew P. Sage Best Transactions Paper Award for the best paper published in all 3 of the IEEE Transactions on Systems, Man, and Cybernetics - Parts A, B, and C in 2006; (2) National University of Singapore Overseas Graduate Scholarship for Ph.D. studies in Carnegie Mellon University (CMU) in 2004-2009; (3) Singapore Computer Society Prize for Best M.Sc. Thesis in School of Computing, National University of Singapore in 2003; and (4) Faculty Teaching Excellence Award in School of Computing, National University of Singapore in 2017-2018. Dr. Low has served as a World Economic Forum’s Global Future Councils Fellow for the Council on the Future of Artificial Intelligence and Robotics from Sep 2016 to Jun 2018 and an IEEE Robotics & Automation Society (RAS) Distinguished Lecturer for the IEEE RAS Technical Committee on Multi-Robot Systems in Mar 2019. He has served as an organizing chair for the IEEE RAS Summer School on Multi-Robot Systems in Jun 2016 and the AI Summer Schools in Jul 2019 and Aug 2020. Dr. Low has also served as associate editors, area chairs and program committee members, and reviewers for premier AI (specifically, multiagent systems, AI planning, robotics, machine learning) conferences: IJCAI, AAAI, ECAI, AAMAS, ICAPS, RSS, IROS, ICRA, CoRL, NeurIPS, ICML, AISTATS, ICLR and journals: TKDE, JMLR, JAIR, MLJ, TNNLS, T-ASE, IJRR, T-RO, AURO, JFR, TOSN, JAAMAS. He was the top 5% reviewer for ICML 2019 and top 33% reviewer for ICML 2020.

Patrick Jaillet (MIT)

More from the Same Authors