Skip to content
/ PICNN Public

PICNN: A Pathway towards Interpretable Convolutional Neural Networks

Notifications You must be signed in to change notification settings

spdj2271/PICNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PICNN

About

This repository is the official PyTorch implementation of "PICNN: A Pathway towards Interpretable Convolutional Neural Networks", Wengang Guo $^*$, Jiayi Yang $^*$, Huilin Yin, Qijun Chen, Wei Ye, AAAI 2024.

Experiment

  • Apply the python environment:

    conda create -n PICNN python=3.8
    conda activate PICNN
    ./requirements.sh
  • Conduct the experiment:

Standard CNNs (STD) using ResNet as the backbone:

python main.py -configFileName='./configs/cifar10.yml' -backbone='resnet18' -criterion='StandardCE'

PICNN:

python main.py -configFileName='./configs/cifar10.yml' -backbone='resnet18' -criterion='ClassSpecificCE'

About

PICNN: A Pathway towards Interpretable Convolutional Neural Networks

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published