Back

Integration of Mechanistic Immunological Knowledge into a Machine Learning Pipeline Increases Predictive Power

Culos, A. E.; Tsai, A. S.; Stanley, N.; Becker, M.; Ghaemi, M. S.; McIlwain, D. R.; Tanada, A.; Fallahzadeh, R.; Nassar, H.; Ganio, E.; Peterson, L.; Han, x.; Stelzer, I.; Ando, K.; Gaudilliere, D.; Phongpreecha, T.; Maric, I.; Chang, A. L.; Shaw, G. M.; Stevenson, D. K.; Bendall, S.; Davis, K.; Fantl, W. C.; Nolan, G. P.; Hastie, T.; Tibshirani, R. j.; Angst, M.; Gaudilliere, B.; Aghaeepour, N.

2020-02-27 bioinformatics
10.1101/2020.02.26.967232 bioRxiv
Show abstract

The dense network of interconnected cellular signaling responses quantifiable in peripheral immune cells provide a wealth of actionable immunological insights. While high-throughput single-cell profiling techniques, including polychromatic flow and mass cytometry, have matured to a point that enables detailed immune profiling of patients in numerous clinical settings, limited cohort size together with the high dimensionality of data increases the possibility of false positive discoveries and model overfitting. We introduce a machine learning platform, the immunological Elastic-Net (iEN), which incorporates immunological knowledge directly into the predictive models. Importantly, the algorithm maintains the exploratory nature of the high-dimensional dataset, allowing for the inclusion of immune features with strong predictive power even if not consistent with prior knowledge. In three independent studies our method demonstrates improved predictive power for clinically-relevant outcomes from mass cytometry data generated from whole blood, as well as a large simulated dataset.

Matching journals

The top 6 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.