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

Curved Projection Reformation

UFPR-ADMR-v2 Dataset

Simple and Robust Loss Design for Multi-Label Learning with Missing Labels

Semantic Image Synthesis with SPADE

Official code release for "Learned Spatial Representations for Few-shot Talking-Head Synthesis" ICCV 2021

Camera Distortion-aware 3D Human Pose Estimation in Video with Optimization-based Meta-Learning

FedMM: Saddle Point Optimization for Federated Adversarial Domain Adaptation

Vit-ImageClassification - Pytorch ViT for Image classification on the CIFAR10 dataset

Adversarial vulnerability of powerful near out-of-distribution detection

MARS: Learning Modality-Agnostic Representation for Scalable Cross-media Retrieva

Representing Long-Range Context for Graph Neural Networks with Global Attention

Code for the Convolutional Vision Transformer (ConViT)

Implementation of ReSeg using PyTorch

Training Confidence-Calibrated Classifier for Detecting Out-of-Distribution Samples / ICLR 2018

Framework web SnakeServer.

🗣️ Microsoft Edge TTS for Home Assistant, no need for app_key

DiffQ performs differentiable quantization using pseudo quantization noise. It can automatically tune the number of bits used per weight or group of weights, in order to achieve a given trade-off between model size and accuracy.

Pre-trained model, code, and materials from the paper "Impact of Adversarial Examples on Deep Learning Models for Biomedical Image Segmentation" (MICCAI 2019).

Second-Order Neural ODE Optimizer, NeurIPS 2021 spotlight

Negative Sample Matters: A Renaissance of Metric Learning for Temporal Grounding