Abstract
Background Polygenic risk scores (PRS) summarise genetic information into a single number with multiple clinical and research uses. Machine learning (ML) has revolutionised a diverse set of fields, however, the impact of ML on genomics in general, and PRSs in particular, has been less significant. We explore how ML can improve the generation of PRSs.
Methods We train ML models on known PRSs using UK Biobank data. We explore whether the models can recreate human programmed PRSs, including using a single model to generate multiple PRSs, and the difficulty in using ML for PRS generation. We also investigate how ML can compensate for missing data and the constraints on performance.
Results We demonstrate almost perfect generation of PRSs, including when using one model to predict multiple scores, and with little loss of performance with reduced quantity of training data. For an example set of missing SNPs the MLP produces predictions that enable separation of cases from population samples with an area under the receiver operating characteristic curve of 0.847 (95% CI: 0.828-0.864) compared to 0.798 (95% CI: 0.779-0.818) for the PRS. We provide evidence that input information is the limiting factor of further improvement.
Conclusions ML can accurately generate PRSs, including with one model for multiple PRSs. The models are transferable and have high longevity. With certain missing SNPs the ML models can statistically significantly improve on normal PRS generation. Models trained are probably at the edge of performance and further improvements likely require use of additional input data.
Code is available at https://github.com/stevensquires/
Competing Interest Statement
RAO and MNW had a UK Medical Research Council (MRC) Confidence in Concept grant to develop a type 1 diabetes GRS biochip with Randox and have ongoing research funding from Randox R D.
Funding Statement
RAO and MNW had a UK Medical Research Council (MRC) Confidence in Concept grant to develop a type 1 diabetes GRS biochip with Randox and have ongoing research funding from Randox R D. SS is funded by a grant from Randox. SS, MNW and RAO are supported by the National Institute for Health and Care Research Exeter Biomedical Research Centre. The views expressed are those of the author(s) and not necessarily those of the NIHR or the Department of Health and Social Care. RAO has research funding from a Diabetes UK Harry Keen Fellowship (16/0005529), National Institute of Diabetes and Digestive and Kidney Diseases grants (NIH R01 DK121843–01 and U01DK127382–01), JDRF grants (3–SRA–2019–827–S–B, 2–SRA–2022–1261–S–B, 2–SRA–2002–1259–S–B, 3–SRA–2022–1241–S–B, and 2–SRA–2022–1258–M–B) and the Larry M. and Leona B. Helmsley Charitable Trust.
Author Declarations
I confirm all relevant ethical guidelines have been followed, and any necessary IRB and/or ethics committee approvals have been obtained.
Yes
The details of the IRB/oversight body that provided approval or exemption for the research described are given below:
The study used only openly available human data from the UK Biobank (https://www.ukbiobank.ac.uk/) and T1DGC (https://repository.niddk.nih.gov/studies/t1dgc/)
I confirm that all necessary patient/participant consent has been obtained and the appropriate institutional forms have been archived, and that any patient/participant/sample identifiers included were not known to anyone (e.g., hospital staff, patients or participants themselves) outside the research group so cannot be used to identify individuals.
Yes
I understand that all clinical trials and any other prospective interventional studies must be registered with an ICMJE-approved registry, such as ClinicalTrials.gov. I confirm that any such study reported in the manuscript has been registered and the trial registration ID is provided (note: if posting a prospective study registered retrospectively, please provide a statement in the trial ID field explaining why the study was not registered in advance).
Yes
I have followed all appropriate research reporting guidelines, such as any relevant EQUATOR Network research reporting checklist(s) and other pertinent material, if applicable.
Yes
Data Availability
All data produced are available online at https://www.ukbiobank.ac.uk/ and https://repository.niddk.nih.gov/studies/t1dgc/
