Skip to content

Analytic score#50

Merged
rousseab merged 26 commits intomainfrom
analytic_score
Jun 3, 2024
Merged

Analytic score#50
rousseab merged 26 commits intomainfrom
analytic_score

Conversation

@rousseab
Copy link
Collaborator

Code to generate some 'fake' harmonic data, as well as some ad hoc mini-experiments to investigate the behavior of various models in the 2 atoms in 3D setting.

Various results and interpretations related to this code are presented in https://www.notion.so/Can-we-compute-an-analytic-approximation-to-the-score-network-ab3780fc19de4af38589e66c2324f182.

The high level conclusions are:

  • "Diffusion MACE", as currently implemented, doesn't seem to learn anything
  • "Vanilla MACE + Equivariant Head" does pretty well, even if the loss doesn't go as low as the theoretical minimum... The samples seem to look pretty good still!

This suggest that we should modify 'Diffusion MACE' to better match the Equivariant Head that was tacked on top of Vanilla MACE... This is work for another day.

@rousseab rousseab requested a review from sblackburn86 May 31, 2024 20:45
rousseab added 2 commits June 3, 2024 11:44
# Conflicts:
#	crystal_diffusion/models/position_diffusion_lightning_model.py
@rousseab rousseab merged commit 560c670 into main Jun 3, 2024
@rousseab rousseab deleted the analytic_score branch June 3, 2024 16:37
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