An Optimal Transport-driven Approach for Cultivating Latent Space in Online Incremental Learning
About
In online incremental learning, data continuously arrives with substantial distributional shifts, creating a significant challenge because previous samples have limited replay value when learning a new task. Prior research has typically relied on either a single adaptive centroid or multiple fixed centroids to represent each class in the latent space. However, such methods struggle when class data streams are inherently multimodal and require continual centroid updates. To overcome this, we introduce an online Mixture Model learning framework grounded in Optimal Transport theory (MMOT), where centroids evolve incrementally with new data. This approach offers two main advantages: (i) it provides a more precise characterization of complex data streams, and (ii) it enables improved class similarity estimation for unseen samples during inference through MMOT-derived centroids. Furthermore, to strengthen representation learning and mitigate catastrophic forgetting, we design a Dynamic Preservation strategy that regulates the latent space and maintains class separability over time. Experimental evaluations on benchmark datasets confirm the superior effectiveness of our proposed method.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Class-incremental learning | CIFAR-100 | -- | 248 | |
| Class-incremental learning | CIFAR-100 | Average Accuracy56.5 | 116 | |
| Class-incremental learning | CIFAR100 (test) | -- | 116 | |
| Online Class-Incremental Learning | Tiny-ImageNet | Average Accuracy39.5 | 60 | |
| Online Class-Incremental Learning | CIFAR-10 (test) | Average Forgetting9.8 | 30 | |
| Online Class-Incremental Learning | CIFAR-10 | Average Accuracy76.1 | 30 | |
| Online Class-Incremental Learning | CIFAR-100 | Average Accuracy56.5 | 30 | |
| Online Class-Incremental Learning | Tiny ImageNet (test) | Average Forgetting16.5 | 30 | |
| Class-incremental learning | CIFAR10 | Average Accuracy87.03 | 12 | |
| Class-incremental learning | MNIST | Average Accuracy97.7 | 8 |