Dual-Teacher Class-Incremental Learning with Data-Free Generative Replay
Abstract
This paper proposes two novel knowledge transfer techniques for class-incremental learning (CIL). First, we propose data-free generative replay (DF-GR) to mitigate catastrophic forgetting in CIL by using synthetic samples from a generative model. In the conventional generative replay, the generative model is pre-trained for old data and shared in extra memory for later incremental learning. In our proposed DF-GR, we train a generative model from scratch without using any training data, based on the pre-trained classification model from the past, so we curtail the cost of sharing pre-trained generative models. Second, we introduce dual-teacher information distillation (DT-ID) for knowledge distillation from two teachers to one student. In CIL, we use DT-ID to learn new classes incrementally based on the pre-trained model for old classes and another model (pre-)trained on the new data for new classes. We implemented the proposed schemes on top of one of the state-of-the-art CIL methods and showed the performance improvement on CIFAR-100 and ImageNet datasets.
Cite
Text
Choi et al. "Dual-Teacher Class-Incremental Learning with Data-Free Generative Replay." IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, 2021. doi:10.1109/CVPRW53098.2021.00393Markdown
[Choi et al. "Dual-Teacher Class-Incremental Learning with Data-Free Generative Replay." IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops, 2021.](https://mlanthology.org/cvprw/2021/choi2021cvprw-dualteacher/) doi:10.1109/CVPRW53098.2021.00393BibTeX
@inproceedings{choi2021cvprw-dualteacher,
title = {{Dual-Teacher Class-Incremental Learning with Data-Free Generative Replay}},
author = {Choi, Yoojin and El-Khamy, Mostafa and Lee, Jungwon},
booktitle = {IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops},
year = {2021},
pages = {3543-3552},
doi = {10.1109/CVPRW53098.2021.00393},
url = {https://mlanthology.org/cvprw/2021/choi2021cvprw-dualteacher/}
}