{"about":{"site":"https://codewithpapers.app","non_affiliation":"Code with Papers and Syntology are not affiliated with, endorsed by, or sponsored by Papers with Code, Meta, or the pwc-archive mirror.","licence":"CC BY-SA 4.0","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","attribution":"https://codewithpapers.app/attribution","modified":"archive material modified by Syntology; see the attribution page"},"url":"/paper/disease-state-prediction-from-single-cell","title":"Disease State Prediction From Single-Cell Data Using Graph Attention Networks","arxiv_id":"2002.07128","date":"2020-02-14","proceeding":null,"authors":["Neal G. Ravindra","Arijit Sehanobish","Jenna L. Pappalardo","David A. Hafler","David van Dijk"],"abstract":"Single-cell RNA sequencing (scRNA-seq) has revolutionized biological discovery, providing an unbiased picture of cellular heterogeneity in tissues. While scRNA-seq has been used extensively to provide insight into both healthy systems and diseases, it has not been used for disease prediction or diagnostics. Graph Attention Networks (GAT) have proven to be versatile for a wide range of tasks by learning from both original features and graph structures. Here we present a graph attention model for predicting disease state from single-cell data on a large dataset of Multiple Sclerosis (MS) patients. MS is a disease of the central nervous system that can be difficult to diagnose. We train our model on single-cell data obtained from blood and cerebrospinal fluid (CSF) for a cohort of seven MS patients and six healthy adults (HA), resulting in 66,667 individual cells. We achieve 92 % accuracy in predicting MS, outperforming other state-of-the-art methods such as a graph convolutional network and a random forest classifier. Further, we use the learned graph attention model to get insight into the features (cell types and genes) that are important for this prediction. The graph attention model also allow us to infer a new feature space for the cells that emphasizes the differences between the two conditions. Finally we use the attention weights to learn a new low-dimensional embedding that can be visualized. To the best of our knowledge, this is the first effort to use graph attention, and deep learning in general, to predict disease state from single-cell data. We envision applying this method to single-cell data for other diseases.","url_abs":"https://arxiv.org/abs/2002.07128v2","url_pdf":"https://arxiv.org/pdf/2002.07128v2.pdf","source":{"archive":"pwc-archive (Hugging Face), CC BY-SA 4.0","snapshot":"2025-07-28","licence_url":"https://creativecommons.org/licenses/by-sa/4.0/legalcode","row_kind":"abstracts"},"code_links":[{"paper_slug":"disease-state-prediction-from-single-cell","repo_url":"https://github.com/vandijklab/scGAT","is_official":1,"mentioned_in_paper":1,"mentioned_in_github":1,"framework":"pytorch","reach":{"status":"ok","spdx":"MIT"}}],"tasks":[{"task_slug":"disease-prediction","task_name":"Disease Prediction"},{"task_slug":"graph-attention","task_name":"Graph Attention"},{"task_slug":"node-classification","task_name":"Node Classification"}],"methods":[],"datasets_introduced":[],"methods_introduced":[],"results":[],"syntology":{"syntology_url":"https://syntology.ai/paper/2002.07128","atlas_url":"https://app.syntology.ai/?focus=2002.07128","mcp":{"get_harvested_code_for_paper":{"arxiv_id":"2002.07128"}},"developers":"https://syntology.ai/developers","read_at":"2026-09-25T09:33:49+00:00","read_at_is":"when the build read Syntology's graph, not when any sample ran","claim":"Per-sample execution status on synthesized fixtures; not a correctness claim about the paper. Samples come from repositories linked to the paper, official or community; repo_kind says which.","repos":[{"provenance":"external:paperswithcode_snapshot_2025-07-28","url":"https://github.com/vandijklab/scGAT","reach":{"status":"ok","spdx":"MIT"}}],"summary":{"ran_honours":1,"ran":2},"by_repo_kind":{"official":{"samples":3,"ran":3,"repositories":1}},"repo_kind_vocabulary":{"official":"The archive marks this repository official for the paper","named_in_paper":"The archive records that the paper mentions this repository; it is not marked official","listed":"In the archive's code links for this paper, not marked official and not recorded as mentioned in the paper","found_in_text":"Syntology found this repository in the paper's own text; whether it is the authors' implementation is not asserted","community":"Not in the archive's code links for this paper; a community repository Syntology harvested"},"n_pointer_only_for_licence":0,"samples":[{"code_sha256_prefix":"18557871f0c93b3f","entry":"accuracy","repo":"vandijklab/scGAT","repo_kind":"official","path":"transduction/load_data.py","file_url":"https://github.com/vandijklab/scGAT/blob/HEAD/transduction/load_data.py","link_basis":"harvester_set","language":"python","status":"ran_honours","verification_level":1,"contract_check":"HONOURS","metamorphic_tier":"invariant","behaviour_fingerprint":true,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"18557871f0c93b3f"}},{"code_sha256_prefix":"eb4a83a8b30f833d","entry":"masked_nll_loss","repo":"vandijklab/scGAT","repo_kind":"official","path":"transduction/load_data.py","file_url":"https://github.com/vandijklab/scGAT/blob/HEAD/transduction/load_data.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"eb4a83a8b30f833d"}},{"code_sha256_prefix":"6226c492214d2a52","entry":"scipysparse2torchsparse","repo":"vandijklab/scGAT","repo_kind":"official","path":"induction/ind.py","file_url":"https://github.com/vandijklab/scGAT/blob/HEAD/induction/ind.py","link_basis":"first_harvest_node","language":"python","status":"ran","verification_level":1,"contract_check":null,"metamorphic_tier":null,"behaviour_fingerprint":false,"licence":"MIT","inline_ok":true,"mcp_get_code":{"code_sha256":"6226c492214d2a52"}}]},"arxiv_metadata":null,"syntology_extracted_results":null}