Skip to content

feat(uq-models): implement deep ensemble and MC dropout models, add UQ visualizations to PostProcessor. - #85

Open
khakis23 wants to merge 33 commits into
developfrom
deep-ensemble
Open

feat(uq-models): implement deep ensemble and MC dropout models, add UQ visualizations to PostProcessor.#85
khakis23 wants to merge 33 commits into
developfrom
deep-ensemble

Conversation

@khakis23

@khakis23 khakis23 commented Jul 2, 2026

Copy link
Copy Markdown
Collaborator

UQ Models Implementation

This feature adds two new UQ model wrappers, deep ensemble (DE) and MC dropout (MCD), which introduce additional visualizations into the post processor.

New Classes:

Key Features:

  • Models integrates with existing model infrastructure (Tuner, PostProcessor, etc.)
  • Negative Log Likelihood loss function is added for neural networks
  • New UQ visualizer class that integrates into Post Processor
    • New uncertainty visualization method wrapper in Post Processor
    • Integrates variance features into 3 existing Post Processor visualization methods automatically

Changes:

@khakis23
khakis23 requested a review from npanczyk July 17, 2026 12:32
@khakis23
khakis23 marked this pull request as ready for review July 17, 2026 12:33
@khakis23
khakis23 marked this pull request as draft July 22, 2026 17:44
@khakis23 khakis23 changed the title feat(deep-ensemble): implement deep ensemble model and tests, add NLL loss function feat(uq-models): implement deep ensemble and MC dropout models, add UQ visualizations to PostProcessor. Aug 1, 2026
@khakis23
khakis23 marked this pull request as ready for review August 14, 2026 01:18
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants