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

classifier guided diffusion #15

Open
tonytu16 opened this issue Nov 12, 2023 · 2 comments
Open

classifier guided diffusion #15

tonytu16 opened this issue Nov 12, 2023 · 2 comments

Comments

@tonytu16
Copy link

In the paper it is mentioned that "An advantage of
our approach is that diffusion can be directly guided by function classifiers that operate in sequence space. We first
sought to guide the network with the DeepGOPlus Gene Ontology (GO) classifier36 to generate proteins with specific
characteristics and functions. Although GO classification scores increased with guidance for nitrogen compound
metabolic process (GO:0006807) and membrane (GO:0016020), we found the classifier had a high false positive rate
often assigning high scores to native sequences outside the GO domain (Figure S10)."

I wonder where is this classifier guidance implemented? I can't seem to find it in the code. If I want to try other classifiers with your code, where should I add the gradients to? Thanks

@0merle0
Copy link
Member

0merle0 commented Nov 13, 2023

in the utils folder there is a potentials.py file where all of the potentials are located, you can incorporate your own classifer into here. In the sample generation loop, after the sequence is passed through the model the potential function is queried. If you need more help implementing let me know and I am happy to help!

@pgmikhael
Copy link

pgmikhael commented Nov 20, 2023

Hi,

Pseudo-code for this would be helpful as well (something akin to what's in the paper), with regards to sequence-level guidance. Thank you!

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

No branches or pull requests

3 participants