Correction to: Nature Machine Intelligence https://doi.org/10.1038/s42256-024-00974-9, published online 16 January 2025.

In the version of the article initially published, in the “COMET deep learning architecture” section of the Methods, the equations for MSE and BSE were switched and have now been corrected as shown:

$${\rm{MSE}}=\frac{1}{N}\mathop{\sum }\limits_{i=1}^{N}\left({y}_{{\rm{true}}}-{y}_{{\rm{pred}}}\right)$$

and

$${\rm{BCE}}=-\frac{1}{N}\mathop{\sum }\limits_{i=1}^{N}\left({y}_{{\rm{true}}}\log \left[{y}_{{\rm{pred}}}\right]+\left(1-{y}_{{\rm{true}}}\right)\log \left[1-{y}_{{\rm{pred}}}\right]\right).$$

Additionally, the grey background circles in Fig. 4d were incorrectly offset from the clusters and have now been corrected. These corrections have been made to the HTML and PDF versions of the article.