Skip to content

Code for the paper "Principled Input-Output-Conditioned Post-Hoc Uncertainty Estimation for Regression Networks" (IO-CUE)

License

Notifications You must be signed in to change notification settings

biggzlar/IO-CUE

Repository files navigation

👑 Input-Output-Conditioned Uncertainty Estimation (IO-CUE) 👑

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

A minimal implementation of the proposed IO-CUE framework, including all experiments shown in the submission.

Getting the data

How to use the project

  • Train a base model using the train_base_model.py script.
  • Train a post-hoc learner using the train_post_hoc_model.py script and a config file. For example: python train_post_hoc_model.py -yc configs/yaml_configs/edgy_depth_gaussian_io_cue.yaml -d 0.

About

Code for the paper "Principled Input-Output-Conditioned Post-Hoc Uncertainty Estimation for Regression Networks" (IO-CUE)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages