Fig. 2: Visualization of the highest-weighted motifs and atoms identified by the model for molecules from the Sider test set containing the SO2NH substructure.

a–c Motifs with the highest attention weights from the motif-level branch. d–f Atoms with the highest attention weights from the atom-level branch.