Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Shallow ensemble as an uncertainty quantification method #57

Merged
merged 11 commits into from
Dec 9, 2024

Conversation

Benzoin96485
Copy link
Owner

  1. Shallow ensemble implementation in the basic DenseLayer, installed as a layer hyperparameter in PhysNet and SpookyNet, with atomic property affine, electrostatic energy, charge conservation, dipole monent calculation, energy reduce, force calculation layers supported.
  2. A highly flexible shallow ensemble reduce layer is added to collect means and variations from ensemble predictions.
  3. Two experimental types of NLL loss based on ensemble means and variations.
  4. "In situ" re-evaluation of test set.
  5. New monitor output format.
  6. Fix the bug that true labels are not inverse-transformed in saved evaluation results.
  7. Saving non-target prediction from the raw output of the model is supported

@Benzoin96485 Benzoin96485 merged commit 31e8b0d into devel Dec 9, 2024
@Benzoin96485 Benzoin96485 linked an issue Dec 9, 2024 that may be closed by this pull request
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.

Implement shallow-ensemble based uncertainty quantification
1 participant