PigeonColumba compressa

Temperature x KD-weight grid

Which temperature T and what mix of KD vs. ground-truth CE (alpha) works best when the student only sees the labeled set?

Methods: Knowledge distillation (Hinton)

Line chart of test accuracy (20-way) against temperature T (log scale), one line per alpha, with a dashed reference line for the teacher. Titled “Temperature x KD-weight grid”. The numbers are in the table below.
Test accuracy (20-way) against temperature T
Line chart of test NLL against temperature T (log scale), one line per alpha. Titled “Temperature x KD-weight grid”. The numbers are in the table below.
Test NLL against temperature T

Table

Mean over the seeds; ± is the standard deviation where more than one seed ran. The experiment took 112 s.

Temperature TAlphaTest accuracy (20-way)Target-class AP ('bird')Agreement with teacherTest NLLKL(Bayes || model)Parameters (nonzero)
10 (labels only)0.664 ± 0.0090.843 ± 0.0050.674 ± 0.0101.018 ± 0.0270.936 ± 0.028820
10.50.663 ± 0.0090.844 ± 0.0050.674 ± 0.0101.018 ± 0.0280.936 ± 0.029820
10.90.664 ± 0.0090.839 ± 0.0030.675 ± 0.0101.017 ± 0.0270.936 ± 0.028820
11.00.663 ± 0.0090.839 ± 0.0030.673 ± 0.0101.018 ± 0.0270.936 ± 0.028820
20 (labels only)0.664 ± 0.0090.843 ± 0.0050.674 ± 0.0101.018 ± 0.0270.936 ± 0.028820
20.50.651 ± 0.0110.841 ± 0.0070.663 ± 0.0111.188 ± 0.0371.105 ± 0.040820
20.90.647 ± 0.0090.839 ± 0.0120.658 ± 0.0101.284 ± 0.0261.201 ± 0.028820
21.00.645 ± 0.0080.839 ± 0.0110.657 ± 0.0101.298 ± 0.0271.215 ± 0.029820
40 (labels only)0.664 ± 0.0090.843 ± 0.0050.674 ± 0.0101.018 ± 0.0270.936 ± 0.028820
40.50.640 ± 0.0100.845 ± 0.0100.652 ± 0.0091.739 ± 0.0621.658 ± 0.061820
40.90.640 ± 0.0090.843 ± 0.0120.652 ± 0.0081.845 ± 0.0671.765 ± 0.068820
41.00.640 ± 0.0090.843 ± 0.0130.651 ± 0.0081.861 ± 0.0691.780 ± 0.070820
80 (labels only)0.664 ± 0.0090.843 ± 0.0050.674 ± 0.0101.018 ± 0.0270.936 ± 0.028820
80.50.622 ± 0.0070.830 ± 0.0260.637 ± 0.0062.048 ± 0.0611.962 ± 0.063820
80.90.612 ± 0.0140.825 ± 0.0240.628 ± 0.0132.203 ± 0.1072.114 ± 0.109820
81.00.610 ± 0.0150.824 ± 0.0220.627 ± 0.0142.227 ± 0.1062.138 ± 0.108820
160 (labels only)0.664 ± 0.0090.843 ± 0.0050.674 ± 0.0101.018 ± 0.0270.936 ± 0.028820
160.50.597 ± 0.0110.815 ± 0.0310.614 ± 0.0132.122 ± 0.1112.033 ± 0.115820
160.90.584 ± 0.0130.811 ± 0.0260.601 ± 0.0132.331 ± 0.1312.239 ± 0.136820
161.00.582 ± 0.0120.811 ± 0.0240.598 ± 0.0122.360 ± 0.1292.268 ± 0.134820

Reference points

Teacher (the large model)
The full-size network trained on the labeled data; the smaller models here are measured against it. Test accuracy (20-way) 0.914, target-class AP ('bird') 0.989, test NLL 0.384, KL(Bayes || model) 0.309, parameters (nonzero) 140,052.
Bayes-optimal (the best any model can do on this problem)
Predictions from the exact class probabilities, which the lab can compute because it generated the data. Test accuracy (20-way) 0.973, target-class AP ('bird') 0.999, test NLL 0.079.

Birds in this experiment

The table's 15 smaller models hatch 1 kind of bird. Each kind is drawn once, for one model: the smallest if the bird is healthy or bones, the one that keeps the most if it's starving. Rows with the same size and score count as one model.

Measured against this experiment's teacher: 140,052 parameters, 4481.7 kbit, accuracy 0.914. Storage is 32 bits a parameter, the lab's rule for a dense fp32 net, which every model drawn here is. Species follows how many times smaller a model is. A bird stays healthy while its model keeps at least 95% of the reference's score, and turns to bones at 60% or less.

These are the lab's recorded results, so they aren't added to anyone's life list. Birds you hatch yourself, in the Coop or from a Field Guide run, are saved to your life list.