Fig. 5: Comparisons of the embedding structures between models with and without the inclusion of targeted PRSs.

The embedding layers immediately before the classification layer of the models were extracted and projected into a 2-dimension space using the UMAP method. Panels on the left side were from models with the inclusion of targeted PRSs (II.SCZ, II.BIP, II.MDD and V.II.4C), and panels on the right side were from models without the inclusion of targeted PRSs (IV.SCZ, IV.BIP, IV.MDD and V.IV.4C). For binary classification models, there were no apparent differences in cluster structures. For the multiclass models, model V.IV.4C did not have a CTRL cluster (green), and BIP became the connecting cluster between the SCZ and MDD clusters.