Hierarchical Utility Calibration for Structured Multiclass Decisions
Organizations: The University of Osaka · The University of Tokyo · RIKEN Center for Advanced Intelligence Project (AIP) · Mila – Quebec AI Institute ·
Abstract
In multiclass probabilistic prediction, Utility Calibration (UC), which focuses auditing on specified utilities, has recently received attention as a way to guarantee downstream decisions while controlling computational and sample requirements. At the same time, some multiclass problems have meaningful label hierarchies that play important roles in medicine and image classification, yet how UC evaluates utility within a hierarchy remains insufficiently understood. We show that the difference between realized utility and predicted mean utility admits an exact decomposition into a sum of contributions from the internal nodes of the label tree. This decomposition shows that positive and negative contributions from different nodes can cancel, and that even when UC is small, the utility errors remaining in parts of the hierarchy need not be small. To address this problem, we propose Hierarchical Utility Calibration (HUC), which evaluates each node contribution before summation while retaining the same target utility, subgroup, and predicted-utility interval. We further provide finite-sample evaluation over all predicted-utility intervals and propose HUC-Boost, which updates only violated internal nodes, with theoretical guarantees for both.
Figures & tables
| iNaturalist | SUPPORT2 | |||||||||
| Method | Acc. | AUC | UC | HUC | Updates | Acc. | AUC | UC | HUC | Updates |
| Baseline | .489 .090 | .900 .108 | .272 .161 | .125 .059 | 0.0 0.0 | .499 .199 | .676 .083 | .167 .178 | .148 .142 | 0.0 0.0 |
| UC-B | .462 .088 | .914 .066 | .045 .018 | .036 .011 | 25.4 24.1 | .575 .051 | .681 .079 | .039 .014 | .050 .031 | 33.9 39.4 |
| HUC-B | .463 .090 | .922 .060 | .075 .032 | .034 .009 | 145.9 180.5 | .575 .045 | .698 .060 | .036 .010 | .035 .009 | 45.2 46.9 |
| Temp. | .489 .090 | .890 .110 | .130 .114 | .092 .076 | 0.0 0.0 | .499 .199 | .686 .072 | .077 .069 | .076 .069 | 0.0 0.0 |
| Temp. HUC | .470 .088 | .930 .050 | .047 .014 | .029 .008 | 43.1 25.9 | .587 .032 | .703 .057 | .032 .006 | .032 .006 | 20.4 22.6 |
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
| Appendix | Content and proof locations |
| A : Formal definitions and tree properties | Defines the probability space, label-tree quantities, subtree mean utilities, and the path-product representation. |
| B : Binary classification and trees | Develops the binary-tree calculations and path decomposition. Appendix B.4 directly proves Eq. ( 5 ) and the UC identity in Section 2.2 . |
| C : General multiway trees | Appendix C.1 proves Eq. ( 7 ); Appendix C.2 proves Theorem 3.1 , Eqs. ( 9 ) and ( 8 ), and the mean-consistency decomposition; Appendix C.3 proves orthogonality, uniqueness, and the Parseval identity. |
| D : Basic properties of HUC | Appendix D.1 proves Proposition 3.3 , Eq. ( 14 ), and the implication from canonical calibration. Appendices D.2 – D.4 give counterexamples concerning conditional branches, utility impact, and node-specific audit populations. |
| E : Proof of HUC-Boost | Appendices E.1 and E.2 analyze Algorithm 1 and prove Theorem 3.5 ; Appendix E.3 gives the ordinary-leaf-Brier comparison; Appendix E.4 proves the path-product results. |
| F : Finite-sample evaluation | Appendix F.1 proves Theorem 3.4 ; Appendix F.3 derives the exact tie-preserving scan; Appendix F.4 derives the exact UC–HUC operation counts. |
| Computational stage | UC | HUC |
|---|---|---|
| 1) Precomputation of the required statistics | ||
| 2) Sorting the predicted utility scores | ||
| 3) Maximization over all intervals |
| Method | Transformation | Parameters learned on calibration data |
|---|---|---|
| Temperature | scalar | |
| Vector | ||
| Dirichlet | , |
| Method | Acc. | AUC | UC | HUC | Updates |
|---|---|---|---|---|---|
| Baseline | .4886 .0895 | .9001 .1082 | .2721 .1615 | .1251 .0595 | 0.00 0.00 |
| UC-B | .4618 .0877 | .9138 .0664 | .0452 .0177 | .0361 .0108 | 25.43 24.07 |
| HUC-B | .4633 .0896 | .9223 .0599 | .0748 .0317 | .0345 .0090 | 145.93 180.47 |
| HUC-B (0.05) | .4853 .0888 | .8971 .1106 | .1837 .1767 | .0815 .0725 | 329.23 270.41 |
| Temp. | .4886 .0895 | .8902 .1097 | .1303 .1143 | .0920 .0756 | 0.00 0.00 |
| Temp. HUC | .4702 .0878 | .9301 .0501 | .0473 .0135 | .0289 .0081 | 43.07 25.92 |
| LR | RF | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Method | Acc. | AUC | UC | HUC | Updates | Acc. | AUC | UC | HUC | Updates |
| Baseline | .5436 .0222 | .9522 .0028 | .2439 .0136 | .1225 .0066 | 0.00 0.00 | .5187 .0081 | .9509 .0027 | .2957 .0089 | .1246 .0089 | 0.00 0.00 |
| UC-B | .4786 .0206 | .9531 .0031 | .0721 .0135 | .0536 .0043 | 9.80 3.42 | .5178 .0069 | .9550 .0027 | .0299 .0037 | .0328 .0067 | 18.80 7.46 |
| Temp. | .5436 .0222 | .9537 .0030 | .0704 .0072 | .0574 .0073 | 0.00 0.00 | .5187 .0081 | .9572 .0022 | .0285 .0043 | .0249 .0049 | 0.00 0.00 |
| Vec. | .5460 .0192 | .9556 .0028 | .0639 .0082 | .0379 .0060 | 0.00 0.00 | .5220 .0155 | .9573 .0021 | .0390 .0184 | .0215 .0053 | 0.00 0.00 |
| Dir. | .5055 .0115 | .9521 .0028 | .1810 .0101 | .0891 .0098 | 0.00 0.00 | .2984 .0157 | .8933 .0067 | .1638 .0076 | .0769 .0067 | 0.00 0.00 |
| LR | RF | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Method | Acc. | AUC | UC | HUC | Updates | Acc. | AUC | UC | HUC | Updates |
| HUC-B | .5094 .0049 | .9550 .0027 | .0876 .0050 | .0354 .0048 | 41.80 7.09 | .5034 .0094 | .9513 .0030 | .0429 .0116 | .0286 .0034 | 79.20 21.49 |
| HUC-B (0.05) | .5286 .0144 | .9559 .0031 | .1124 .0198 | .0446 .0081 | 400.20 62.97 | .5196 .0063 | .9544 .0021 | .0410 .0081 | .0206 .0033 | 385.20 70.12 |
| Temp. HUC | .5181 .0093 | .9560 .0025 | .0584 .0114 | .0336 .0090 | 46.00 31.02 | .5070 .0116 | .9564 .0028 | .0315 .0072 | .0204 .0007 | 29.60 22.13 |
| Vec. HUC | .5406 .0104 | .9566 .0028 | .0551 .0109 | .0349 .0055 | 17.80 14.86 | .5226 .0122 | .9571 .0022 | .0298 .0024 | .0232 .0073 | 12.00 12.90 |
| Dir. HUC | .5055 .0115 | .9521 .0028 | .1810 .0101 | .0891 .0098 | 0.00 0.00 | .2984 .0157 | .8933 .0067 | .1638 .0076 | .0769 .0067 | 0.00 0.00 |
| Method | Acc. | AUC | UC | HUC | Updates |
|---|---|---|---|---|---|
| Baseline | .4990 .1994 | .6763 .0828 | .1671 .1777 | .1481 .1422 | 0.00 0.00 |
| UC-B | .5746 .0506 | .6813 .0789 | .0394 .0139 | .0504 .0313 | 33.90 39.40 |
| HUC-B | .5754 .0453 | .6985 .0600 | .0364 .0101 | .0351 .0089 | 45.20 46.89 |
| HUC-B (0.05) | .5023 .1967 | .6844 .0787 | .1466 .1828 | .1270 .1444 | 198.80 148.38 |
| Temp. | .4990 .1994 | .6865 .0722 | .0769 .0692 | .0757 .0693 | 0.00 0.00 |
| Temp. HUC | .5872 .0319 | .7031 .0568 | .0324 .0064 | .0323 .0062 | 20.40 22.61 |
| LR | RF | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Method | Acc. | AUC | UC | HUC | Updates | Acc. | AUC | UC | HUC | Updates |
| Baseline | .6088 .0125 | .7621 .0218 | .0385 .0075 | .0385 .0075 | 0.00 0.00 | .6152 .0053 | .7462 .0289 | .0473 .0083 | .0473 .0083 | 0.00 0.00 |
| UC-B | .6088 .0120 | .7614 .0210 | .0255 .0019 | .0255 .0019 | 3.80 2.17 | .6152 .0128 | .7436 .0289 | .0350 .0104 | .0350 .0104 | 7.60 10.41 |
| Temp. | .6088 .0125 | .7621 .0216 | .0284 .0055 | .0268 .0067 | 0.00 0.00 | .6152 .0053 | .7442 .0284 | .0407 .0083 | .0407 .0083 | 0.00 0.00 |
| Vec. | .6094 .0119 | .7633 .0217 | .0269 .0035 | .0269 .0035 | 0.00 0.00 | .6140 .0038 | .7286 .0394 | .0306 .0071 | .0306 .0071 | 0.00 0.00 |
| Dir. | .6100 .0120 | .7677 .0182 | .0251 .0028 | .0251 .0028 | 0.00 0.00 | .6124 .0087 | .7475 .0133 | .0310 .0095 | .0310 .0095 | 0.00 0.00 |
| LR | RF | |||||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Method | Acc. | AUC | UC | HUC | Updates | Acc. | AUC | UC | HUC | Updates |
| HUC-B | .6086 .0121 | .7580 .0208 | .0282 .0019 | .0271 .0017 | 12.80 7.16 | .6100 .0166 | .7459 .0397 | .0342 .0108 | .0342 .0107 | 19.20 12.77 |
| HUC-B (0.05) | .6112 .0138 | .7608 .0197 | .0285 .0034 | .0285 .0034 | 72.80 54.21 | .6156 .0109 | .7485 .0314 | .0332 .0095 | .0332 .0095 | 120.00 90.77 |
| Temp. HUC | .6098 .0117 | .7597 .0200 | .0303 .0052 | .0303 .0052 | 9.60 15.13 | .6158 .0097 | .7499 .0330 | .0306 .0070 | .0318 .0067 | 16.80 17.30 |
| Vec. HUC | .6088 .0112 | .7555 .0230 | .0296 .0062 | .0278 .0038 | 8.00 10.20 | .6162 .0117 | .7281 .0297 | .0305 .0077 | .0301 .0082 | 16.80 12.46 |
| Dir. HUC | .6098 .0118 | .7641 .0171 | .0256 .0033 | .0256 .0033 | 0.80 1.79 | .6138 .0097 | .7267 .0230 | .0314 .0071 | .0316 .0071 | 20.00 12.33 |
| Target class | Utilities | Relevant nodes | HUC error after correction | Updates | ||||
|---|---|---|---|---|---|---|---|---|
| UC-B | HUC-B | HUC-B (0.05) | UC-B | HUC-B | HUC-B (0.05) | |||
| 3 | 1 | .0333 .0083 | .0343 .0081 | .1264 .1451 | 11.57 11.41 | 11.90 11.79 | 47.77 36.81 | |
| 6 | 3 | .0557 .0431 | .0346 .0081 | .1268 .1448 | 29.30 34.74 | 32.70 35.19 | 138.60 108.89 | |
| 9 | 4 | .0504 .0313 | .0351 .0089 | .1270 .1444 | 33.90 39.40 | 45.20 46.89 | 198.80 148.38 | |
| Method | Acc. | AUC | UC | HUC | Updates |
|---|---|---|---|---|---|
| Baseline | .8222 .0025 | .9913 .0008 | .1380 .0039 | .1190 .0035 | 0.00 0.00 |
| UC-B | .8223 .0026 | .9899 .0009 | .0216 .0077 | .0142 .0043 | 44.60 33.25 |
| HUC-B | .8197 .0036 | .9891 .0009 | .0302 .0078 | .0170 .0065 | 251.60 50.97 |
| HUC-B (0.05) | .8223 .0027 | .9897 .0009 | .0213 .0083 | .0140 .0071 | 395.80 66.34 |
| Temp. | .8222 .0025 | .9906 .0007 | .0335 .0075 | .0189 .0040 | 0.00 0.00 |
| Temp. HUC | .8206 .0029 | .9904 .0008 | .0371 .0057 | .0181 .0048 | 109.40 110.03 |
| Method | Acc. | AUC | UC | HUC | Updates |
|---|---|---|---|---|---|
| Baseline | .9504 .1004 | .9577 .0653 | .0471 .1011 | .0452 .0980 | 0.00 0.00 |
| UC-B | .9816 .0244 | .9687 .0444 | .0039 .0034 | .0035 .0031 | 62.07 39.19 |
| HUC-B | .9801 .0283 | .9672 .0514 | .0039 .0031 | .0033 .0026 | 92.37 69.38 |
| HUC-B (0.05) | .9505 .1004 | .9621 .0605 | .0448 .1014 | .0432 .0983 | 85.37 75.29 |
| Temp. | .9504 .1004 | .9619 .0563 | .0285 .0555 | .0275 .0552 | 0.00 0.00 |
| Temp. HUC | .9838 .0195 | .9766 .0363 | .0036 .0028 | .0032 .0026 | 64.43 66.43 |
| Method | Acc. | AUC | UC | HUC | Updates |
|---|---|---|---|---|---|
| Baseline | .7852 .1560 | .9703 .0382 | .1021 .0998 | .0672 .0633 | 0.00 0.00 |
| UC-B | .7960 .1262 | .9720 .0300 | .0338 .0221 | .0324 .0197 | 31.50 30.98 |
| HUC-B | .7987 .1254 | .9736 .0307 | .0357 .0212 | .0278 .0186 | 35.27 28.58 |
| HUC-B (0.05) | .7960 .1281 | .9749 .0283 | .0415 .0332 | .0325 .0316 | 137.33 80.47 |
| Temp. | .7852 .1560 | .9711 .0396 | .0415 .0354 | .0391 .0364 | 0.00 0.00 |
| Temp. HUC | .8010 .1264 | .9764 .0265 | .0326 .0239 | .0265 .0189 | 24.80 33.13 |
Probability Calibration
Probability calibration measures and improves whether predicted probabilities match observed outcome frequencies, supporting reliable decisions under uncertainty.
See all 134 papers