Skip to content

Pytorch implementation of DoctorNet from "Who Said What: Modeling Individual Labelers Improves Classification"

Notifications You must be signed in to change notification settings

seunghyukcho/doctornet-pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

DoctorNet Pytorch Implementation

Unofficial pytorch implementation of DoctorNet from "Who Said What: Modeling Individual Labelers Improves Classification"

Paper: https://arxiv.org/abs/1703.08774

Experiments

We used LabelMe for validating implementation. LabelMe is an image classification task that was labeled by 77 annotators in AMT(Amazon Mechanical Turk). Original data is here. However in the real data, 18 of the annotators didn't labeled any image. So total 59 annotators' labels were used for training. You can download the preprocessed version here.

Training

Training is done in two-stage. First, we train annotator classifiers with shared feature extractor - Inception v3. After the classifiers converged, we fix them and train weights used for averaging decisions of annotators.

Annotator classifier

Weights

Testing

Model DoctorNet(paper)[1] doctornet-pytorch
Accuracy 82.12 77.61

참조문헌

[1] https://arxiv.org/abs/2012.13052

About

Pytorch implementation of DoctorNet from "Who Said What: Modeling Individual Labelers Improves Classification"

Resources

Stars

Watchers

Forks

Languages