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

Rainbow is all you need! A step-by-step tutorial from DQN to Rainbow

Visual Tracking by TridenAlign and Context Embedding

Laplacian Score-regularized Concrete Autoencoders

📚 A collection of Jupyter notebooks for learning and experimenting with OpenVINO 👓

StyleGAN2-ADA - Official PyTorch implementation

Replication Package for "An Empirical Study of the Effectiveness of an Ensemble of Stand-alone Sentiment Detection Tools for Software Engineering Datasets"

Source code for "FastBERT: a Self-distilling BERT with Adaptive Inference Time".

This repository contains the official code of the paper Equivariant Subgraph Aggregation Networks (ICLR 2022)

MLSpace: Hassle-free machine learning & deep learning development

NFT-Price-Prediction-CNN - Using visual feature extraction, prices of NFTs are predicted via CNN (Alexnet and Resnet) architectures.

Plenoxels: Radiance Fields without Neural Networks, Code release WIP

Facebook AI Image Similarity Challenge: Descriptor Track

code for paper"A High-precision Semantic Segmentation Method Combining Adversarial Learning and Attention Mechanism"

DeLag: Detecting Latency Degradation Patterns in Service-based Systems

Official code for UnICORNN (ICML 2021)

TensorFlow 101: Introduction to Deep Learning for Python Within TensorFlow

official implemntation for "Contrastive Learning with Stronger Augmentations"

The implementation our EMNLP 2021 paper "Enhanced Language Representation with Label Knowledge for Span Extraction".

Phonetic PosteriorGram (PPG)-Based Voice Conversion (VC)

Beyond imagenet attack (accepted by ICLR 2022) towards crafting adversarial examples for black-box domains.