Skip to content

Training a neural network to predict the errors made by a separate network.

Notifications You must be signed in to change notification settings

MKowal2/supervisor-error-guide

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

14 Commits
 
 
 
 
 
 

Repository files navigation

Intermediate Supervisors for Confidence Evaluation and Error-Guided Learning

The first goal of this project was to train a neural network (supervisor) to predict the error that another neural network (student) would make during the task of semantic segmentation. The student network is the DeepLab v2 network from https://github.com/speedinghzl/Pytorch-Deeplab.

Pytorch >= 0.4.0 is required! The rest of the necessary modules can be found in the first cell of MKowal_Project.ipynb.

RESULTS

The supervisor is trained on three types of error: binary class independant, binary class dependant, and multi-class. It achieves a maximum accuracy of over 28.4% meanIOU. The error guided learning is currently being worked on with no state-of-the-art results yet...

Acknowledgments

Done at Ryerson University, Toronto. Under the supervision of Kosta Derpanis (https://www.scs.ryerson.ca/kosta/) and Neil Bruce (https://www.scs.ryerson.ca/~bruce/)

About

Training a neural network to predict the errors made by a separate network.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published