Downloads · 30 days
0
yongers/GKD
GKD is a image segmentation model from yongers. Use it for the image segmentation task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
<div align="center" <img src="fig/logo.png" width="400px" height="400px" </div
Downloads · 30 days
0
Access
Public
Updated Mar 12, 2026
Repo size
24.1 GB
Likes
0
Public
Click a slice to open those files.
.pth24.1 GB · 100%
From the Hugging Face model README
[!Note] This repository contains model weights after domain-general distillation and task learning.
Motivation: Conventional knowledge distillation approaches primarily preserve in-domain accuracy while neglecting out-of-domain generalization, which is essential under distribution shifts. This limitation becomes more severe with the emergence of vision foundation models (VFMs): although VFMs exhibit strong robustness on unseen data, distilling them with conventional KD often compromises this ability.
Methodology: GKD decouples representation learning from task learning. In the first stage, the student acquires domain-agnostic representations through selective feature distillation, and in the second stage, these representations are frozen for task adaptation, thereby mitigating overfitting to visible domains. To further support transfer, we introduce a query-based soft distillation mechanism, where student features act as queries to teacher representations to selectively retrieve transferable spatial knowledge from VFMs. Extensive experiments on five domain generalization benchmarks demonstrate that GKD consistently outperforms existing KD methods, achieving average gains of +1.9% in foundation-to-foundation (F2F) and +10.6% in foundation-to-local (F2L) distillation.