Grounding Predictive Reliability through Masked Observability
Organizations: University of Kassel, Germany
Abstract
An uncertainty estimate is meaningful only if its semantics can be tied to observable predictive behavior. Normal-Inverse-Gamma (NIG) evidential regression provides a structured decomposition of predictive variability into data-scale and predictive-mean components, but marginal likelihood does not identify their allocation. We address this semantic ambiguity in masked representation learning. MORSE (Masked Observability for Reliable Structured Encoding) predicts the same hidden target under controlled changes in visible context and uses normalized cross-context variation to ground predictive-mean uncertainty, while a Gaussian objective learns the conditional mean and data scale. The normalized variation plays the role of the NIG factor , which the marginal leaves unidentified; this yields a policy-relative observability variance that measures dependence on visible evidence, distinct from variation induced by resampling the training set. On a Gaussian-process benchmark with a contextual reference built from analytic Bayes conditional means, MORSE has the lowest log-variance error among compared evidential, ensemble, and dropout estimators in all nine tested settings, while the estimator closest to training-set variation perturbs the model rather than its context in every setting, so the two references separate the estimators empirically. MORSE amortizes observability estimation into one forward pass per observation state. On a heteroscedastic pulse benchmark, its predictor's cross-context variance approaches a large-sample reference as training data increase. MORSE also improves low-label transfer on PTB-XL and in-distribution trajectory prediction on IMPTC.
Figures & tables
| LogVar MAE | ||
|---|---|---|
| Method | Contextual | Training set |
| DER | ||
| RN-DER | ||
| Gaussian-MV | ||
| Deep ensemble (5) | ||
| Bootstrap ensemble (5) | ||
| Method | AUROC | Macro-F1 | ECE | AURC |
|---|---|---|---|---|
| Supervised | .827 .005 | .592 .011 | .069 .010 | .110 .011 |
| DAE | .826 .006 | .596 .012 | .079 .016 | .096 .008 |
| MAE | .842 .006 | .620 .005 | .043 .020 | .099 .008 |
| PatchTST | .823 .011 | .590 .015 | .030 .016 | .120 .014 |
| SimMTM | .849 .005 | .622 .008 | .043 .013 | .090 .007 |
| ST-MEM | .835 .010 | .604 .012 | .052 .020 | .106 .012 |
Appendix figures & tables35 assets
Supplementary material from the paper’s appendix.
Appendix
| Setting | Synthetic | PTB-XL | IMPTC |
|---|---|---|---|
| Views | 4 | 4 | 4 |
| Mask rate per view / shared-target rate | .50 / .25 | .50 / .25 | .50 / .25 |
| / | 1 / 0 | 1 / 0 | 1 / 0 |
| / | .01 / .05 | .05 / .05 | .05 / .05 |
| ObsRef margin / upper weight | .25 / 1 | 1 / .5 | 1 / .5 |
| Target-floor quantile | 0 | .10 | .10 |
| Method | MSE | NLL | |||
|---|---|---|---|---|---|
| DER | .0498 .0008 | .1483 | 2.2526 1.1065 | 4.4823 2.2005 | 6.1481 2.0911 |
| RN-DER | .0488 .0020 | .1641 | 1.3639 .9395 | 2.7457 1.8722 | 5.0433 1.7440 |
| Gaussian-MV | .0362 .0163 | .1258 | 1.4991 1.0410 | 3.0724 2.0066 | 7.4254 2.1064 |
| Method | MSE | NLL | |||
|---|---|---|---|---|---|
| DER | .0591 .0282 | .3317 | .9234 .7862 | 1.8467 1.5562 | 3.2412 1.7635 |
| RN-DER | .0573 .0283 | .3310 | 1.0276 .8912 | 2.0126 1.7041 | 3.9321 1.7180 |
| Gaussian-MV | .0422 .0283 | .3587 | 1.2887 1.0253 | 2.5494 1.9830 | 5.5316 2.7207 |
| Method | MSE | NLL | Err. AUROC | Sel. excess | Cov. gap | ||
|---|---|---|---|---|---|---|---|
| Masked DER | .6188 .1811 | .0516 | 1.4947 .2270 | .5562 .4302 | .8271 .0228 | .0895 .0470 | .0460 .0166 |
| RN-DER | .5151 .0588 | .0268 | 1.1061 .2679 | .3149 .4609 | .8319 .0087 | .0360 .0040 | .0512 .0085 |
| Gaussian-MV | .5416 .0423 | .0499 | .5343 .2257 | .5772 .1008 | .8785 .0168 | .0289 .0025 | .0934 .0020 |
| MORSE | .5331 .0371 | .0298 | .3424 .0906 | .1253 .1092 | .8705 .0070 | .0295 .0034 | .0176 .0086 |
| Predictive fit | Mean uncertainty | Error ranking | Intervals | ||||
| Method | MSE | NLL | AUROC | SelEx | CovGap | Width † | |
| Model comparison | |||||||
| Masked DER | |||||||
| RN-DER | |||||||
| Gaussian-MV | |||||||
| MORSE | |||||||
| Variant | MSE | NLL | Err. AUROC | Sel. excess | Cov. gap | ||
|---|---|---|---|---|---|---|---|
| MORSE | .5331 .0371 | .0298 | .3424 .0906 | .1253 .1092 | .8705 .0070 | .0295 .0034 | .0176 .0086 |
| MORSE without ObsRef | .5440 .0408 | 1.4182 .1172 | .8044 .1085 | .2531 .1233 | .8664 .0095 | .0364 .0040 | .1000 .0001 |
| MORSE without Rank | .5411 .0450 | .0338 | .0114 .0050 | .3191 .1597 | .8737 .0117 | .0292 .0032 | .0768 .0040 |
| Variant | MSE | NLL | AUROC | Sel. | Cov. gap | Width | ||
|---|---|---|---|---|---|---|---|---|
| base | .5474 .0266 | .0406 | .1650 .0167 | .2632 .0519 | .8732 .0081 | .0293 .0027 | .0779 .0041 | 2.0594 .1027 |
| Condition only | .5384 .0393 | 2.5312 .1932 | .4731 .3790 | .8549 .3924 | .8360 .0171 | .0729 .0319 | .1000 .0000 | 39.4471 10.3392 |
| Metric | MORSE-RC | MORSE |
|---|---|---|
| MSE | .5320 .0198 | .5618 .0787 |
| NLL | .0376 | .0509 |
| Error AUROC | .8721 .0082 | .8692 .0125 |
| Selective excess | .0281 .0018 | .0302 .0044 |
| Coverage gap | .0762 .0035 | .0188 .0100 |
| Interval width | 2.0153 .1033 | 1.4804 .0708 |
| MSE | NLL | Error AUROC | train s/run | passes | |
|---|---|---|---|---|---|
| 2 | .5691 .0510 | .0246 | .8739 .0080 | 134.5 8.9 | 3 |
| 4 | .5331 .0371 | .0298 | .8705 .0070 | 213.1 4.8 | 5 |
| 8 | .5200 .0286 | .0588 | .8693 .0080 | 422.2 38.7 | 9 |
| Head | MSE | NLL | Err. AUROC | Sel. | Cov. gap | Width |
|---|---|---|---|---|---|---|
| NIG-shaped | .5331 | .8705 | .0295 | .0176 | 1.4485 | |
| Explicit | .5140 | .8543 | .0402 | .0094 | 1.4328 |
| Method | Center | Log-var | W1 | Cov. gap | Width |
|---|---|---|---|---|---|
| DER | .2562 .0146 | 3.8535 .0897 | .3757 .0333 | .0323 .0023 | 6.1212 .5051 |
| RN-DER | .2307 .0132 | 4.2056 .1369 | .4418 .0269 | .0471 .0012 | 8.3378 .5677 |
| Gaussian-MV | .2354 .0814 | 4.6828 .1083 | .5190 .0526 | .0480 .0008 | 10.5144 1.4901 |
| MORSE | .1584 .0182 | 1.5496 .0513 | .0785 .0086 | .1203 .1085 | 1.3480 .1253 |
| Baseline | Center benefit [95% CI] | Log-var benefit [95% CI] | W1 benefit [95% CI] |
|---|---|---|---|
| DER | .0978 [.0776,.1186] | 2.3039 [2.1914,2.3870] | .2972 [.2631,.3273] |
| RN-DER | .0723 [.0548,.0898] | 2.6560 [2.5830,2.7289] | .3632 [.3473,.3794] |
| Gaussian-MV | .0770 [.0011,.1515] | 3.1332 [3.0243,3.2521] | .4405 [.4047,.4840] |
| Variant | LogVar | Spearman | Center RMSE | NLL | CovGap | Width |
|---|---|---|---|---|---|---|
| NIG C0B0O0 | 2.0910 .1165 | .5932 .0074 | .1395 .0018 | .1702 .0214 | .0806 .0031 | 1.4128 .0485 |
| NIG C0B0O1 | 2.0641 .0798 | .6246 .0041 | .1416 .0047 | .1701 .0159 | .0809 .0022 | 1.3938 .0273 |
| NIG C0B1O0 | .7135 .0081 | .5878 .0070 | .1425 .0056 | .0601 .0090 | .0291 .0045 | 1.0448 .0168 |
| NIG C0B1O1 | .7313 .0118 | .6215 .0034 | .1408 .0061 | .0595 .0107 | .0348 .0042 | 1.0669 .0191 |
| NIG C1B0O0 | 1.9900 .1392 | .5933 .0065 | .1434 .0049 | .1497 .0302 | .0747 .0046 | 1.3445 .0588 |
| Variant | MSE | NLL | AUROC | AURC | CovGap | Width | |
|---|---|---|---|---|---|---|---|
| NIG C0B0O0 | .5137 .0326 | .0714 | .8715 .0074 | .0543 .0030 | .0694 .0062 | 2.4342 .1679 | .1608 .0765 |
| NIG C0B0O1 | .5183 .0366 | .0416 | .8792 .0079 | .0553 .0043 | .0803 .0050 | 2.1985 .2129 | .2314 .1294 |
| NIG C0B1O0 | .5115 .0294 | .0419 | .8748 .0114 | .0534 .0030 | .0149 .0090 | 1.4983 .0932 | .2773 .1708 |
| NIG C0B1O1 | .5143 .0242 | .0664 | .8758 .0064 | .0542 .0025 | .0191 .0080 | 1.4646 .0627 | .1402 .0827 |
| NIG C1B0O0 | .5265 .0181 | .0445 | .8761 .0088 | .0557 .0026 | .0707 .0069 | 2.5693 .3079 | .1911 .1393 |
| Endpoint | Control | Difference [95% interval] | Wins | Holm |
|---|---|---|---|---|
| GP (LogVar) | no Condition | 0.00356 [ 0.00845, 0.01383] | 1/5 | 1.000 |
| GP (LogVar) | no Bound | 1.21951 [ 1.32962, 1.11433] | 5/5 | 0.375 |
| GP (LogVar) | no Order | 0.02515 [ 0.01646, 0.03384] | 0/5 | 0.375 |
| GP (LogVar) | explicit head | 0.00994 [ 0.00080, 0.01802] | 1/5 | 0.750 |
| GP (LogVar) | fixed | 0.00411 [ 0.00470, 0.01234] | 1/5 | 1.000 |
| GP (LogVar) | direct mean variance | 0.00379 [ 0.00625, 0.01383] | 2/5 | 1.000 |
| vs LogVar | vs LogVar | Mean | Mean | |
|---|---|---|---|---|
| 512 | 1.987 .187 | 4.477 | .009 | .0427 |
| 2048 | 1.562 .092 | 3.327 | .022 | .0445 |
| 8192 | 1.281 .091 | 2.345 | .038 | .0361 |
| 32768 | .917 .087 | 1.687 | .040 | .0200 |
| A. Mean | A. LogVar MAE to | |||||||
|---|---|---|---|---|---|---|---|---|
| DER | RN-DER | Gaussian-MV | MORSE | DER | RN-DER | Gaussian-MV | MORSE | |
| 512 | 26.21 7.65 | 28.90 5.43 | 3.72 1.61 | .084 .022 | 6.371 .153 | 7.166 .139 | 3.517 .245 | 1.708 .061 |
| 1024 | 411.22 227.15 | 307.81 236.56 | 1.542 .351 | .0479 .0065 | 7.719 .529 | 6.315 1.161 | 3.893 .137 | 1.564 .079 |
| 2048 | .897 .252 | 1.293 .340 | 1.297 .286 | .0249 .0069 | 3.850 .085 | 4.186 .111 | 4.641 .119 | 1.546 .040 |
| 4096 | .815 .180 | .991 .202 | 1.585 .302 | .0159 .0032 | 4.069 .122 | 4.434 .061 | 5.759 .226 | 1.505 .038 |
| 8192 | 5.50 1.39 | 4.21 .77 | 1.592 .202 | .0140 .0033 | 4.452 .073 | 4.788 .051 | 6.421 .118 | 1.223 .030 |
| Method | Context LogVar | Context | LogVar | |
|---|---|---|---|---|
| DER | ||||
| RN-DER | ||||
| Gaussian-MV | ||||
| MORSE | ||||
| MORSE-IS | ||||
| Deep ensemble (5) |
| Method | Center RMSE | Context LogVar | Context | LogVar | |
|---|---|---|---|---|---|
| DER | |||||
| RN-DER | |||||
| Gaussian-MV | |||||
| Deep ensemble (5) | |||||
| Bootstrap ensemble (5) | |||||
| MC dropout (50) |
| Method | Variance ratio | Coverage | CovGap 95 | Width ratio |
|---|---|---|---|---|
| DER | ||||
| RN-DER | ||||
| Gaussian-MV | ||||
| Deep ensemble (5) | ||||
| Bootstrap ensemble (5) | ||||
| MC dropout (50) |
| (A) Contextual reference | ||||||
|---|---|---|---|---|---|---|
| Noise | MORSE | MC dropout | Bootstrap | [95% interval] | Wins | |
| .025 | Step | 5/5 | ||||
| .025 | Window-homosc. | 5/5 | ||||
| .025 | Smooth | 5/5 | ||||
| .05 | Step (default) | 5/5 | ||||
| .05 | Window-homosc. | 5/5 | ||||
| (A) Contextual reference | |||||
|---|---|---|---|---|---|
| Noise | All | No attn. | MORSE | [95% interval] | |
| .025 | Step | ||||
| .025 | Window-homosc. | ||||
| .025 | Smooth | ||||
| .05 | Step (default) | ||||
| .05 | Window-homosc. | ||||
| Labels | Metric | Benefit | Paired 95% CI | Wins |
|---|---|---|---|---|
| MORSE versus SimMTM | ||||
| 5% | AUROC | .0169 | [.0121, .0208] | 5/5 |
| AUPRC | .0277 | [.0251, .0308] | 5/5 | |
| Macro-F1 | .0147 | [.0090, .0223] | 5/5 | |
| Brier | .0026 | [ .0043, .0096] | 2/5 | |
| AURC | .0009 | [ .0078, .0109] | 2/5 | |
| Method | AUROC | AUPRC | Macro-F1 | Brier | ECE | AURC |
|---|---|---|---|---|---|---|
| 5% labeled records | ||||||
| Supervised | .808 .005 | .587 .010 | .578 .004 | .166 .003 | .067 .016 | .120 .003 |
| DAE | .809 .011 | .588 .019 | .569 .017 | .170 .010 | .111 .036 | .110 .010 |
| MAE | .835 .002 | .628 .012 | .610 .005 | .152 .006 | .046 .021 | .105 .005 |
| PatchTST | .809 .009 | .587 .021 | .575 .007 | .165 .010 | .026 .013 | .128 .014 |
| SimMTM | .835 .005 | .634 .006 | .607 .009 | .151 .005 | .049 .019 | .098 .008 |
| Method | Accuracy | Macro-F1 | Balanced | Directional F1 | Brier | ECE | AURC |
|---|---|---|---|---|---|---|---|
| Supervised | .7154 .0029 | .6362 .0178 | .6479 .0109 | .2951 .0552 | .3751 .0009 | .0243 .0074 | .1435 .0010 |
| DAE | .7240 .0027 | .6565 .0074 | .6622 .0052 | .3392 .0215 | .3539 .0019 | .0211 .0007 | .1309 .0016 |
| Masked AE | .7304 .0027 | .6645 .0049 | .6693 .0032 | .3539 .0161 | .3459 .0026 | .0171 .0034 | .1284 .0030 |
| PatchTST-style | .7363 .0015 | .6742 .0150 | .6783 .0102 | .3697 .0503 | .3346 .0059 | .0244 .0064 | .1193 .0049 |
| SimMTM-style | .7338 .0018 | .6563 .0049 | .6655 .0030 | .3152 .0175 | .3414 .0024 | .0248 .0056 | .1244 .0020 |
| MORSE | .7407 .0018 | .6882 .0087 | .6878 .0066 | .4084 .0339 | .3273 .0036 | .0319 .0038 | .1081 .0034 |
| Method | Accuracy | Macro-F1 | Balanced | Directional F1 | Brier | ECE | AURC |
|---|---|---|---|---|---|---|---|
| Supervised | .6194 .0495 | .4714 .0422 | .5352 .0430 | .0030 .0046 | .4956 .0657 | .0435 .0175 | .2688 .0781 |
| DAE | .7134 .0029 | .6022 .0149 | .6311 .0059 | .1711 .0506 | .3822 .0025 | .0212 .0038 | .1621 .0044 |
| Masked AE | .7199 .0032 | .6202 .0140 | .6418 .0067 | .2196 .0455 | .3731 .0023 | .0237 .0099 | .1569 .0029 |
| PatchTST-style | .7256 .0052 | .6158 .0095 | .6418 .0068 | .1899 .0264 | .3708 .0077 | .0249 .0088 | .1568 .0084 |
| SimMTM-style | .7198 .0025 | .6022 .0090 | .6335 .0041 | .1570 .0264 | .3807 .0024 | .0346 .0066 | .1656 .0032 |
| MORSE | .7225 .0084 | .6129 .0122 | .6425 .0078 | .1891 .0203 | .3761 .0131 | .0411 .0224 | .1595 .0126 |
| Split | Baseline | Metric | Oriented [95% CI] | Wins/5 | |
|---|---|---|---|---|---|
| In-dist. | Masked AE | Macro-F1 | 5 | 2.30 | |
| In-dist. | Masked AE | Brier | 5 | 3.41 | |
| In-dist. | Masked AE | ECE | 0 | ||
| In-dist. | Masked AE | AURC | 5 | 3.62 | |
| In-dist. | PatchTST-style | Macro-F1 | 5 | 1.57 | |
| In-dist. | PatchTST-style | Brier | 3 | .85 |
| Experiment | Model/data seeds | Fixed auxiliary seeds |
|---|---|---|
| Fixed and parameterized one-dimensional tasks | 10000–10049 | fixed-generator settings in merge manifests |
| Pulse-window comparison/components/ | 1–10 | paired masks and data within each seed |
| MORSE versus MORSE-RC | 1–20 | paired data, initialization, and masks |
| Strict five-anchor reference | anchors 900001–900005; reference 1–50 | test 424242; normalization 777777; init/mask/eval/loader 12001–12004 |
| Strict sensitivity | anchors 900001–900010; reference 1–50 at each | same 424242, 777777, and 12001–12004 |
| Pulse proxy reference | standard 700011–700018; large 700021–700024 | 128-mask bank 424243 |
| Claim | Primary evidence | Scope |
|---|---|---|
| Marginal fitting does not identify the relative allocation | Proposition 1 | Concerns evidence from the marginal, not uniqueness under arbitrary regularization |
| Cross-context mean variance is observable and has a population limit | Propositions 2 and 3 | Policy-relative; not a parameter posterior |
| The learned score aligns with the contextual reference | GP audit, Tables 1 , 21 , 22 , and 23 | Four-context score; scale error in nine GP settings, five anchors each, not ordering (MC dropout has the higher Spearman correlation in five); MC dropout close at length scale .10; undercoverage; benchmark-dependent (Table 20 ) |
| The learned score costs fewer inference passes than direct repeated masking of comparable accuracy | GP amortization audit, Appendix E | Default setting; four passes versus 16 to 128; inference only, not training cost |
| Contextual and training-set variation are distinct | GP two-reference audit in nine settings (Table 23 ); Proposition 3 | MORSE–bootstrap reversal in all nine GP settings; disabling attention dropout moves MC dropout toward (Table 24 ), by refitting rather than an inference-only switch; not observed on the pulse benchmark |
| The predictor’s contextual response converges with | Pulse training-size audit, Table 18 | Approximate proxy reference, not exact Bayes |
Uncertainty Quantification
Uncertainty quantification estimates how reliable predictions and scientific inferences are, and how uncertainty propagates into decisions.
See all 660 papers