Skip to content
/ DTL Public

This repository is the official implementation of "DTL: Disentangled Transfer Learning for Visual Recognition", which is accepted by AAAI 2024.

Notifications You must be signed in to change notification settings

heekhero/DTL

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 

Repository files navigation

Disentangled Transfer Learning (DTL)

Implementation of DTL, a parameter-efficient paradigm that disentangles trainable parameters from a backbone model via a Compact Side Network (CSN). This repository supports both image classification and object detection tasks.


๐Ÿ“– Paper

Minghao Fu, Ke Zhu, Jianxin Wu
DTL: Disentangled Transfer Learning for Visual Recognition, arxiv โ€ข AAAI Paper


๐Ÿš€ Features

  • Memory-efficient: significantly reduces GPU memory footprint by offloading trainable parameters to CSN.
  • Parameter-efficient: only a small fraction of parameters (LoRA-style) is fine-tuned.
  • Accurate: achieves state-of-the-art performance on ImageNet, VTAB, and COCO benchmarks.
  • Versatile: easy extension to image classification and object detection frameworks.
  • Seamless: integrates with PyTorch and MMDetection pipelines.

๐Ÿ› ๏ธ Usage

  • Classification: see classification/README.md for training and evaluation steps.
  • Detection: see detection/README.md for COCO training and evaluation instructions.

๐ŸŽ“ Citation

If you use DTL in your work, please cite:

@inproceedings{fu_dtl2024,
author = {Fu, Minghao and Zhu, Ke and Wu, Jianxin},
title = {{DTL}: disentangled transfer learning for visual recognition},
year = {2024},
booktitle = {Proceedings of the Thirty-Eighth AAAI Conference on Artificial Intelligence},
pages={12082 - 12090}
}

About

This repository is the official implementation of "DTL: Disentangled Transfer Learning for Visual Recognition", which is accepted by AAAI 2024.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published