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

Official PyTorch Implementation of SSMix (Findings of ACL 2021)

This app is a simple example of using Strealit to create a financial data web app.

Official PyTorch implementation of MAAD: A Model and Dataset for Attended Awareness

A Multi-modal Perception Tracker (MPT) for speaker tracking using both audio and visual modalities

This repository contains the data and code for the paper "Diverse Text Generation via Variational Encoder-Decoder Models with Gaussian Process Priors" ([email protected])

Code for STFT Transformer used in BirdCLEF 2021 competition.

HuSpaCy: industrial-strength Hungarian natural language processing

Code for Universal Semi-Supervised Semantic Segmentation models paper accepted in ICCV 2019

General purpose Slater-Koster tight-binding code for electronic structure calculations

El-Gamal on Elliptic Curve (Python)

Defending graph neural networks against adversarial attacks (NeurIPS 2020)

[NeurIPS 2020] Code for the paper "Balanced Meta-Softmax for Long-Tailed Visual Recognition"

Laser device for neutralizing - mosquitoes, weeds and pests

Byzantine-robust decentralized learning via self-centered clipping

A pure PyTorch batched computation implementation of "CIF: Continuous Integrate-and-Fire for End-to-End Speech Recognition"

Red Team tool for exfiltrating files from a target's Google Drive that you have access to, via Google's API.

City Surfaces: City-scale Semantic Segmentation of Sidewalk Surfaces

VOneNet: CNNs with a Primary Visual Cortex Front-End

Code for Subgraph Federated Learning with Missing Neighbor Generation (NeurIPS 2021)

Codes and pretrained weights for winning submission of 2021 Brain Tumor Segmentation (BraTS) Challenge