ACEsuit / mace

MACE - Fast and accurate machine learning interatomic potentials with higher order equivariant message passing.
Other
413 stars 157 forks source link

need to be able to specify what head is used for _prediction_ with a multihead model #390

Open bernstei opened 2 months ago

bernstei commented 2 months ago

Whether this is best done at the time of fitting (by masking out the pretrained head) or when creating the MACECalculator, there needs to be some way to clearly specify which head is used for prediction. The former is perhaps better for the use case of stabilizing fine tuning, but the latter is a bit more general

gabor1 commented 2 months ago

It might be interesting to always use the latter (but it should default to the new head), because one would be interested in checking the stability of the new model on the entire periodic table. This also needs to have the elements not in the new training set to be dropped only for the new head.

-- Gábor

Gábor Csányi Professor of Molecular Modelling Engineering Laboratory, University of Cambridge Pembroke College Cambridge

Pembroke College supports CARA. A Lifeline to Academics at Risk. http://www.cara.ngo/

On 23 Apr 2024, at 19:51, bernstei @.***> wrote:

Whether this is best done at the time of fitting (by masking out the pretrained head) or when creating the MACECalculator, there needs to be some way to clearly specify which head is used for prediction. The former is perhaps better for the use case of stabilizing fine tuning, but the latter is a bit more general

— Reply to this email directly, view it on GitHubhttps://github.com/ACEsuit/mace/issues/390, or unsubscribehttps://github.com/notifications/unsubscribe-auth/ABIANZ7OYHCAIJIK2BWIMY3Y62UTZAVCNFSM6AAAAABGVPWMU6VHI2DSMVQWIX3LMV43ASLTON2WKOZSGI2TSNJVHA4DMMY. You are receiving this because you are subscribed to this thread.Message ID: @.***>