Skip to content

Analytic score #50

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

Merged
merged 26 commits into from
Jun 3, 2024
Merged

Analytic score #50

merged 26 commits into from
Jun 3, 2024

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
1 check passed
@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