Cycle class consistency with distributional optimal transport and knowledge distillation for unsupervised domain adaptation

Tuan Nguyen, Van Nguyen, Trung Le, He Zhao, Quan Hung Tran, Dinh Phung

Research output: Chapter in Book/Report/Conference proceedingConference PaperResearchpeer-review

8 Citations (Scopus)

Abstract

Unsupervised domain adaptation (UDA) aims to transfer knowledge from a model trained on a labeled source domain to an unlabeled target domain. To this end, we propose in this paper a novel cycle class-consistent model based on optimal transport (OT) and knowledge distillation. The model consists of two agents, a teacher and a student cooperatively working in a cycle process under the guidance of the distributional optimal transport and distillation manner. The OT distance is designed to bridge the gap between the distribution of the target data and a distribution over the source class-conditional distributions. The optimal probability matrix then provides pseudo labels to learn a teacher that achieves a good classification performance on the target domain. Knowledge distillation is performed in the next step in which the teacher distills and transfers its knowledge to the student. And finally, the student produces its prediction for the optimal transport step. This process forms a closed cycle in which the teacher and student networks are simultaneously trained to conduct transfer learning from the source to the target domain. Extensive experiments show that our proposed method outperforms existing methods, especially the class-aware and OT-based ones on benchmark datasets including Office-31, Office-Home, and ImageCLEF-DA.

Original languageEnglish
Title of host publicationProceedings of the 38th Conference on Uncertainty in Artificial Intelligence, UAI 2022
EditorsJames Cussens, Kun Zhang
Place of PublicationLondon UK
PublisherProceedings of Machine Learning Research (PMLR)
Pages1519-1529
Number of pages11
Volume180
ISBN (Electronic)9781713863298
Publication statusPublished - 2022
EventConference in Uncertainty in Artificial Intelligence 2022 - Eindhoven, Netherlands
Duration: 1 Aug 20225 Aug 2022
Conference number: 38th
https://proceedings.mlr.press/v180/ (Proceedings)
https://www.auai.org/uai2022/ (Website)

Conference

ConferenceConference in Uncertainty in Artificial Intelligence 2022
Abbreviated titleUAI 2022
Country/TerritoryNetherlands
CityEindhoven
Period1/08/225/08/22
Internet address

Cite this