where each term represents loss for 𝑡 = 𝑇, 𝑡 -1, 0, respectively. To further enhance the model's ability to accurately reconstruct the atom types, an auxiliary cross-entropy loss term 𝐿 ;-is added, which directly penalizes the discrepancy between the predicted atom types and the true atom types at time step 𝑡 = 0.