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

Regularizers #114

Merged
merged 27 commits into from
Jan 9, 2025
Merged

Regularizers #114

merged 27 commits into from
Jan 9, 2025

Conversation

rousseab
Copy link
Collaborator

@rousseab rousseab commented Jan 8, 2025

I implement a Regularizer class to try to regularize the score. I run and analyze some small experiments in the experiments folder.

I try various ideas, which are documented in this note: https://www.notion.so/Regularizing-the-Relative-Coordinates-Score-Network-174690083be48092addad80dda88a58b

Mostly for validating that a score network has enough capacity to learn the ground truth.
Copy link
Collaborator

@sblackburn86 sblackburn86 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

A minor question to look at. Rest looks good.


if end_index == 0:
end_time = 0.0
end_sigma = 0.0
Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

is this correct? I believe sigma(t=0) is sigma_min, not zero.

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

considering this is used to compute an "effective" sigma based on start and end sigma, it should not affect the results much.

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Unfortunately, the last two values of sigma_t are identical in the sigma_array. If I set end_sigma = sigma_min, I'll invoke the conditional score method with sigma = 0, which goes poopoo. To avoid this, I set end_sigma to zero here. As you noted, it shouldn't really change anything.

@rousseab rousseab merged commit 4871158 into main Jan 9, 2025
1 check passed
@rousseab rousseab deleted the fokker_planck_regularizer branch January 9, 2025 20:23
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