UDON: Universal Dynamic Online distillatioN for generic image representations

Universal image representations are critical in enabling real-world fine-grained and instance-level recognition applications, where objects and entities from any domain must be identified at large scale. Despite recent advances, existing methods fail to capture important domain-specific knowledge, while also ignoring differences in data distribution across different domains. This leads to a large performance gap between efficient universal solutions and expensive approaches utilising a collection of specialist models, one for each domain. In this work, we make significant strides towards closing this gap, by introducing a new learning technique, dubbed UDON (Universal Dynamic Online DistillatioN). UDON employs multi-teacher distillation, where each teacher is specialized in one domain, to transfer detailed domain-specific knowledge into the student universal embedding. UDON's distillation approach is not only effective, but also very efficient, by sharing most model parameters between the student and all teachers, where all models are jointly trained in an online manner. UDON also comprises a sampling technique which adapts the training process to dynamically allocate batches to domains which are learned slower and require more frequent processing. This boosts significantly the learning of complex domains which are characterised by a large number of classes and long-tail distributions. With comprehensive experiments, we validate each component of UDON, and showcase significant improvements over the state of the art in the recent UnED benchmark. Code: https://github.com/nikosips/UDON .

Paper

Similar papers

Peer review

Reviewer uq4A3/10 · confidence 4/52024-07-11

Summary

The paper introduces a novel method for enhancing universal image embeddings through a multi-teacher knowledge distillation approach. The method, named UDON, employs a dynamic sampling technique and a shared backbone across multiple domain-specific teachers and a universal student model to efficiently distill domain-specific knowledge into a universal embedding. This embedding model can work as a foundation for many downstream task models, such as classification, retrieval, generation, etc., with a high potential impact on the community.

Strengths

1) The introduction of multi-teacher distillation with a shared backbone is novel and addresses significant challenges in universal image representation. (2) The dynamic sampling method that adapts based on domain-specific performance enhances learning efficiency and addresses the imbalance in training data distribution. (3) Extensive experiments demonstrate that UDON outperforms existing methods, showcasing the effectiveness of the proposed techniques.

Weaknesses

(1) This paper misses a series of very important works for embedding learning [1,2]. The [1] proposed the Matryoshka representation learning (MRL) that the OpenAI's recent embedding model adopts. MRL would unfold the embedding model into multiple dimensions such as 64, 128, 512 or more. This paper simply takes the 64 and cannot be extended to higher dimensions which significantly limited its potential in applications. [1] Kusupati, Aditya, et al. "Matryoshka representation learning." Advances in Neural Information Processing Systems 35 (2022): 30233-30249. [2] Cai, Mu, et al. "Matryoshka Multimodal Models." arXiv preprint arXiv:2405.17430 (2024). (2) Tab 1 is pretty confusing. There are "Off-the-shelf", "Specialist+Oracle", "ImageNet21k pretraining" and "CLIP pretraining". It is hard to understand the exact meaning of them. Moreover, the evaluation is not convincing. It misses some recent SOTA embedding models, such as SigLIP. (3) Please unify the format of subtitles. In Line 253, there is "Implementation Details" but line 235 applies "Compared methods". (4) There is no ablation study over the temperature and embedding dimension, which should be critical hyperparameters to explore.

Questions

N/A

Rating

3

Confidence

4

Soundness

2

Presentation

3

Contribution

2

Limitations

N/A

Reviewer uq4A2024-08-10

After Rebuttal

Thank you for the detailed response from the authors. I will maintain my original score, as the novelty and impact of the work seem limited. The concept of multi-teacher distillation is not new, and testing solely on UnED with fixed dim size may not be sufficiently convincing.

Reviewer xwWg6/10 · confidence 3/52024-07-12

Summary

The paper proposes Universal Dynamic Online distillatioN (UDON), which is a multi-teacher distillation method designed for universal image representations. UDON adopts a knowledge distillation strategy by distilling information from multiple teacher model trained for different domains to a student model to learn the universal embedding. It also proposes a dynamic domain sampling strategy for balancing the different domains. The provided experimental results verify its effectiveness.

Strengths

The proposed design is simple and elegant. The authors describe the design and implementation in details, which makes it easy to follow. The experimental results also demonstrate its effectiveness. In addition, many ablation experiments are conducted to give insight. I suppose the paper is valuable enough to be accepted.

Weaknesses

**The reason why the proposed method is better than the previous work (USC) is not straightforward and clear enough for me.** In the paper, the main baseline is Universal Separate Classifier Training method (USC). It uses a backbone to learn an universe embedding where multiple classifier heads for different domains are trained separately with the universe embedding as input. Compared with USC, UDON introduces distillation and dynamic domain sampling. I understand the part about dynamic domain sampling but feel confused about the distillation. As shown in Figure 2, UDON uses multiple extra classifier heads as teachers for distillation. Since the teacher heads and student head are all based on the previous embedding $E_b$, why the distillation strategy can help to build an better universal embedding $E_u$? The authors explain the reason mainly on [Line 37 - Line 51] and [Line 164 - Line 169], while I am still confused about it. On [Line 37 - Line 51], it says that "it is difficult to encode detailed knowledge about many image domains in a single model". Therefore, the authors propose to use knowledge distillation between a student and teachers model for different domains, while the student and teachers share the backbone. Does it mean that the difficulty remains for USC, but it is solved by UDON with the distillation? The story seems a bit conflicting to me. Because from my point of view, a backbone with more classifier heads is still a single model, since the heads are mostly very light-weight compared with the backbone. In my opinion, the proposed distillation design is filtering the information from $E_b$, which keeps more useful and universal information for the student head to learn $E_u$.

Questions

See weakness.

Rating

6

Confidence

3

Soundness

3

Presentation

2

Contribution

3

Limitations

See weakness.

Reviewer CCei6/10 · confidence 4/52024-07-13

Summary

The paper tackles the problem of multi-domain fine-grained instance recognition/retrieval. The authors propose to train a unified backbone for all modalities with online distillation with domain-specific teachers, improving the performance compared to naive single backbone baselines and being competitive with expensive methods that utilize a number of domain-specific specialists. The authors propose a number of technical contributions in addition to the online distillation, including dynamic resampling based on a proxy of the task difficulty. UDON exhibits favorable performance compared to a number of baselines on the UnED benchmark consisting of a high number of fine-grained domain-specific benchmarks.

Strengths

- The paper is very well written with clarity and sufficient details about the implementations and the intuitive explanation of the main contributions. - UDON exhibits improvements over strong baselines for a variety of tasks. - The proposed dynamic domain sampling is interesting and achieves a clear boost to the performance for challenging benchmarks like Met and GLDv2. - The paper includes many ablations about the different design choices which helps in understanding where the gains stem from.

Weaknesses

- [important] The paper tackles generalization for instance recognition/retrieval systems, but the scaling axis of this question has not been studied. This is lacking since in the past few years we have witnessed scalable pre-training in terms of data and parameters being a very effective solution to many generalization problems. - The performance gains provided by UDON seem highly sensitive to which dataset is used for the evaluation. For example, while the dynamic sampling contribution helps two datasets, it dropped the performance of the other five. This does not mean that this contribution should be dismissed but rather we might need more work to achieve its objectives without hurting the performance of other stable benchmarks. - Some ablations of the projections architecture as well as the design of the MLP baseline would be a welcome addition.

Questions

1) depending on the number of classes the magnitude of the loss can be very different. How is this handled in equation [5]? 2) an important question is whether scaling is indeed the answer for most problems. For the off-the-shelf baselines, it can be easy to test much larger backbones (e.g. DinoV2-G, Clip-Large, …). It is interesting to see if simply scaling general-purpose pre-training would be sufficient to address the issues tackled in the paper. 3) Related to the question above, given that the performance gain between a naive unified model vs UDON is significant but not enormous, I wonder if scaling the naive single model baseline’s capacity slightly would be sufficient to address the shortcomings of training one model with many fine-grained domains.

Rating

6

Confidence

4

Soundness

3

Presentation

4

Contribution

3

Limitations

Yes

Reviewer Mmnc5/10 · confidence 3/52024-07-17

Summary

The paper proposes an online distillation approach in a multi-teacher setup w/ weight sharing for efficiency. A strategic dynamic batch sampling process has been proposed to help domains w/ slower learning during training.

Strengths

The unified backbone and batch sampling strategy is novel and powerful. The paper is well written and experiments/datasets carefully chosen.

Weaknesses

Did not spot any, though, it seems there is a throughput drop.

Questions

Can you elaborate more on impact of the throughput drop in real-world industry applications? Also, specify how practical applications could benefit - please provide specific details.

Rating

5

Confidence

3

Soundness

3

Presentation

3

Contribution

3

Limitations

Mentioned in paper

Reviewer CCei2024-08-13

I would like to thank the authors for the rebuttal, it has addressed the majority of my questions/concerns. Therefore, I will raise my score to 6.

Reviewer xwWg2024-08-14

Thanks for your detailed response. After reading the response and other reviewers' comments, I will keep my original score.

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC