Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/josephkj/iod
(TPAMI 2021) iOD: Incremental Object Detection via Meta-Learning
https://github.com/josephkj/iod
continual-learning detectron2 incremental-learning meta-learning object-detection pami-2021 tpami
Last synced: 15 days ago
JSON representation
(TPAMI 2021) iOD: Incremental Object Detection via Meta-Learning
- Host: GitHub
- URL: https://github.com/josephkj/iod
- Owner: JosephKJ
- License: apache-2.0
- Created: 2021-10-19T06:45:59.000Z (about 3 years ago)
- Default Branch: main
- Last Pushed: 2022-12-19T09:09:15.000Z (almost 2 years ago)
- Last Synced: 2024-03-14T20:21:37.759Z (8 months ago)
- Topics: continual-learning, detectron2, incremental-learning, meta-learning, object-detection, pami-2021, tpami
- Language: Python
- Homepage: https://josephkj.in
- Size: 1.46 MB
- Stars: 104
- Watchers: 4
- Forks: 19
- Open Issues: 8
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
## Incremental Object Detection via Meta-Learning
#### Published in IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI)
##### DOI 10.1109/TPAMI.2021.3124133
Early access on IEEE Xplore: [https://ieeexplore.ieee.org/document/9599446](https://ieeexplore.ieee.org/document/9599446)
arXiv paper: [https://arxiv.org/abs/2003.08798](https://arxiv.org/abs/2003.08798)
## Abstract
In a real-world setting, object instances from new classes can be continuously encountered by object detectors. When existing object detectors are applied to such scenarios, their performance on old classes deteriorates significantly. A few efforts have been reported to address this limitation, all of which apply variants of knowledge distillation to avoid catastrophic forgetting.
We note that although distillation helps to retain previous learning, it obstructs fast adaptability to new tasks, which is a critical requirement for incremental learning. In this pursuit, we propose a meta-learning approach that learns to reshape model gradients, such that information across incremental tasks is optimally shared. This ensures a seamless information transfer via a meta-learned gradient preconditioning that minimizes forgetting and maximizes knowledge transfer. In comparison to existing meta-learning methods, our approach is task-agnostic, allows incremental addition of new-classes and scales to high-capacity models for object detection.
We evaluate our approach on a variety of incremental learning settings defined on PASCAL-VOC and MS COCO datasets, where our approach performs favourably well against state-of-the-art methods.
Figure: Qualitative results of our incremental object detector trained in a 10+10 setting where the first task contain instances of aeroplane, bicycle, bird, boat, bottle, bus, car, cat, chair and cow, while the second task learns instance from diningtable, dog, horse, motorbike, person, pottedplant, sheep, sofa, train and tvmonitor. Our model is able to detect instances of both tasks alike, without forgetting.
## Installation and setup
- Install the Detectron2 library that is packages along with this code base. See [INSTALL.md](INSTALL.md).
- Download and extract Pascal VOC 2007 to `./datasets/VOC2007/`
- Use the starter script: `run.sh`
## Trained Models and Logs
| Setting | Reported mAP | Reproduced mAP | Commands | Models and logs |
|:-------:|:------------:|:--------------:|:--------:|:---------------:|
| 19+1 | 70.2 | 70.4 | [run.sh](https://github.com/JosephKJ/iOD/blob/main/run.sh#L1-L8) | [Google Drive](https://drive.google.com/file/d/1sW-aZ9crRFjgbErtgXNQ8hO67WLKYAAn/view?usp=sharing) |
| 15+5 | 67.8 | 69.6 | [run.sh](https://github.com/JosephKJ/iOD/blob/main/run.sh#L11-L19) | [Google Drive](https://drive.google.com/file/d/1E8m4VrrKmNYT1Zba0MwaI3ZjztrLobcA/view?usp=sharing) |
| 10+10 | 66.3 | 67.3 | [run.sh](https://github.com/JosephKJ/iOD/blob/main/run.sh#L22-L30) | [Google Drive](https://drive.google.com/file/d/1LH7OY-uMifl2gwCFEgm6U5h_Xfh1nPcH/view?usp=sharing) |
##### Configurations with which the above results were reproduced:
- Python version: 3.6.7
- PyTorch version: 1.3.0
- CUDA version: 11.0
- GPUs: 4 x NVIDIA GTX 1080-ti
## Acknowledgement
The code is build on top of Detectron2 library.
## Citation
If you find our research useful, please consider citing us:
```BibTeX
@ARTICLE {joseph2021incremental,
author = {Joseph. KJ and Jathushan. Rajasegaran and Salman. Khan and Fahad. Khan and Vineeth. N Balasubramanian},
journal = {IEEE Transactions on Pattern Analysis & Machine Intelligence},
title = {Incremental Object Detection via Meta-Learning},
year = {2021},
issn = {1939-3539},
doi = {10.1109/TPAMI.2021.3124133},
publisher = {IEEE Computer Society},
address = {Los Alamitos, CA, USA},
month = {nov}
}
```