A Machine Teaching Framework for Scalable Recognition

Last update: Dec 08, 2021

Related tags

Overview

MEMORABLE

This repository contains the source code accompanying our ICCV 2021 paper.

A Machine Teaching Framework for Scalable Recognition
Pei Wang, Nuno Vasconcelos.
In ICCV, 2021.

@InProceedings{wang2021gradient,
author = {Wang, Pei and Vasconcelos, Nuno},
title = {A Machine Teaching Framework for Scalable Recognition},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2021}
}

Requirements

The project was implemented and tested in Python 3.5 and Pytorch 1.0. Other versions should work after minor modification.
NVIDIA GPU and cuDNN are required to have fast speeds. For now, CUDA 8.0 with cuDNN 6.0.20 has been tested. The other versions should be working.

Datasets

Butterflies and Chinese Characters, Gull are used. Please organize them as below after download,

datasets
|_ butterflies_crop
  |_ images
    |_ Viceroy
    |_ ...
|_ chinese_chars
  |_ images
    |_ grass
    |_ ...
|_ CUBgull
  |_ images
    |_ CaliforniaGull
    |_ ...

Implementation details

To generate counterfactual explanations

get_all_CE_butterflies.py
get_all_CE_gull.py

by SimCLR model

get_all_CE_butterflies_simclr.py
get_all_CE_gull_simclr.py

To generate teaching images and explanations

train_butterflies_CMaxGrad.py
train_gull_CMaxGrad.py

by SimCLR model

train_butterflies_CMaxGrad_simclr.py
train_gull_CMaxGrad_simclr.py

For the SimCLR, we used repo for training. Thanks for their open-source. For standard model, we trained by

train_butterflies_resnet18.py
train_gull_resnet18_linear.py

For questions, feel free to reach out

Pei Wang: [email protected]

A Machine Teaching Framework for Scalable Recognition

Related tags

Overview

MEMORABLE

Requirements

Datasets

Implementation details

To generate counterfactual explanations

To generate teaching images and explanations

Owner

BoxInst: High-Performance Instance Segmentation with Box Annotations

[SIGGRAPH Asia 2019] Artistic Glyph Image Synthesis via One-Stage Few-Shot Learning

Western-3DSlicer-Modules - Point-Set Registrations for Ultrasound Probe Calibrations

Automatic differentiation with weighted finite-state transducers.

This repository contains a PyTorch implementation of the paper Learning to Assimilate in Chaotic Dynamical Systems.

AirLoop: Lifelong Loop Closure Detection

Using deep learning to predict gene structures of the coding genes in DNA sequences of Arabidopsis thaliana

Remote sensing change detection tool based on PaddlePaddle

Implementation of ConvMixer in TensorFlow and Keras

git《Investigating Loss Functions for Extreme Super-Resolution》(CVPR 2020) GitHub:

A simple rest api serving a deep learning model that classifies human gender based on their faces. (vgg16 transfare learning)

PixelPick This is an official implementation of the paper "All you need are a few pixels: semantic segmentation with PixelPick."

Domain Adaptation with Invariant RepresentationLearning: What Transformations to Learn?

Official repository of "BasicVSR++: Improving Video Super-Resolution with Enhanced Propagation and Alignment"

An easy-to-use app to visualise attentions of various VQA models.

I-BERT: Integer-only BERT Quantization

Implementation of QuickDraw - an online game developed by Google, combined with AirGesture - a simple gesture recognition application

This repository contains the source codes for the paper AtlasNet V2 - Learning Elementary Structures.

This is an early in-development version of training CLIP models with hivemind.

PyTorch implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation.