{"id":"https://openalex.org/W4408061109","doi":"https://doi.org/10.14778/3705829.3705837","title":"NeutronTP: Load-Balanced Distributed Full-Graph GNN Training with Tensor Parallelism","display_name":"NeutronTP: Load-Balanced Distributed Full-Graph GNN Training with Tensor Parallelism","publication_year":2024,"publication_date":"2024-10-01","ids":{"openalex":"https://openalex.org/W4408061109","doi":"https://doi.org/10.14778/3705829.3705837"},"language":"en","primary_location":{"id":"doi:10.14778/3705829.3705837","is_oa":false,"landing_page_url":"https://doi.org/10.14778/3705829.3705837","pdf_url":null,"source":{"id":"https://openalex.org/S4210226185","display_name":"Proceedings of the VLDB Endowment","issn_l":"2150-8097","issn":["2150-8097"],"is_oa":false,"is_in_doaj":false,"is_core":true,"host_organization":"https://openalex.org/P4310319798","host_organization_name":"Association for Computing Machinery","host_organization_lineage":["https://openalex.org/P4310319798"],"host_organization_lineage_names":["Association for Computing Machinery"],"type":"journal"},"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the VLDB Endowment","raw_type":"journal-article"},"type":"article","indexed_in":["crossref"],"open_access":{"is_oa":false,"oa_status":"closed","oa_url":null,"any_repository_has_fulltext":false},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5112330510","display_name":"Xin Ai","orcid":"https://orcid.org/0009-0006-0746-8222"},"institutions":[{"id":"https://openalex.org/I9224756","display_name":"Northeastern University","ror":"https://ror.org/03awzbc87","country_code":"CN","type":"education","lineage":["https://openalex.org/I9224756"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Xin Ai","raw_affiliation_strings":["Northeastern University, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Northeastern University, China","institution_ids":["https://openalex.org/I9224756"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5085132219","display_name":"Hao Yuan","orcid":"https://orcid.org/0000-0002-5973-846X"},"institutions":[{"id":"https://openalex.org/I9224756","display_name":"Northeastern University","ror":"https://ror.org/03awzbc87","country_code":"CN","type":"education","lineage":["https://openalex.org/I9224756"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Hao Yuan","raw_affiliation_strings":["Northeastern University, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Northeastern University, China","institution_ids":["https://openalex.org/I9224756"]}]},{"author_position":"middle","author":{"id":null,"display_name":"Zeyu Ling","orcid":null},"institutions":[{"id":"https://openalex.org/I9224756","display_name":"Northeastern University","ror":"https://ror.org/03awzbc87","country_code":"CN","type":"education","lineage":["https://openalex.org/I9224756"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Zeyu Ling","raw_affiliation_strings":["Northeastern University, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Northeastern University, China","institution_ids":["https://openalex.org/I9224756"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5052641511","display_name":"Qiange Wang","orcid":"https://orcid.org/0000-0002-4847-6070"},"institutions":[{"id":"https://openalex.org/I165932596","display_name":"National University of Singapore","ror":"https://ror.org/01tgyzw49","country_code":"SG","type":"education","lineage":["https://openalex.org/I165932596"]}],"countries":["SG"],"is_corresponding":false,"raw_author_name":"Qiange Wang","raw_affiliation_strings":["National University of Singapore, Singapore"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"National University of Singapore, Singapore","institution_ids":["https://openalex.org/I165932596"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100367496","display_name":"Yanfeng Zhang","orcid":"https://orcid.org/0000-0002-9871-0304"},"institutions":[{"id":"https://openalex.org/I9224756","display_name":"Northeastern University","ror":"https://ror.org/03awzbc87","country_code":"CN","type":"education","lineage":["https://openalex.org/I9224756"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Yanfeng Zhang","raw_affiliation_strings":["Northeastern University, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Northeastern University, China","institution_ids":["https://openalex.org/I9224756"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5113100564","display_name":"Zhenbo Fu","orcid":null},"institutions":[{"id":"https://openalex.org/I9224756","display_name":"Northeastern University","ror":"https://ror.org/03awzbc87","country_code":"CN","type":"education","lineage":["https://openalex.org/I9224756"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Zhenbo Fu","raw_affiliation_strings":["Northeastern University, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Northeastern University, China","institution_ids":["https://openalex.org/I9224756"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5107249760","display_name":"Chaoyi Chen","orcid":"https://orcid.org/0009-0002-9921-4326"},"institutions":[{"id":"https://openalex.org/I9224756","display_name":"Northeastern University","ror":"https://ror.org/03awzbc87","country_code":"CN","type":"education","lineage":["https://openalex.org/I9224756"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Chaoyi Chen","raw_affiliation_strings":["Northeastern University, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Northeastern University, China","institution_ids":["https://openalex.org/I9224756"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5086475357","display_name":"Yu Gu","orcid":"https://orcid.org/0000-0003-4345-6932"},"institutions":[{"id":"https://openalex.org/I9224756","display_name":"Northeastern University","ror":"https://ror.org/03awzbc87","country_code":"CN","type":"education","lineage":["https://openalex.org/I9224756"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Yu Gu","raw_affiliation_strings":["Northeastern University, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Northeastern University, China","institution_ids":["https://openalex.org/I9224756"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5072406974","display_name":"Ge Yu","orcid":"https://orcid.org/0000-0002-3171-8889"},"institutions":[{"id":"https://openalex.org/I9224756","display_name":"Northeastern University","ror":"https://ror.org/03awzbc87","country_code":"CN","type":"education","lineage":["https://openalex.org/I9224756"]}],"countries":["CN"],"is_corresponding":false,"raw_author_name":"Ge Yu","raw_affiliation_strings":["Northeastern University, China"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Northeastern University, China","institution_ids":["https://openalex.org/I9224756"]}]}],"institutions":[],"countries_distinct_count":2,"institutions_distinct_count":9,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":2.1383,"has_fulltext":false,"cited_by_count":7,"citation_normalized_percentile":{"value":0.89685253,"is_in_top_1_percent":false,"is_in_top_10_percent":false},"cited_by_percentile_year":{"min":97,"max":99},"biblio":{"volume":"18","issue":"2","first_page":"173","last_page":"186"},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T11273","display_name":"Advanced Graph Neural Networks","score":0.9990000128746033,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},"topics":[{"id":"https://openalex.org/T11273","display_name":"Advanced Graph Neural Networks","score":0.9990000128746033,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10028","display_name":"Topic Modeling","score":0.9968000054359436,"subfield":{"id":"https://openalex.org/subfields/1702","display_name":"Artificial Intelligence"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10036","display_name":"Advanced Neural Network Applications","score":0.9926999807357788,"subfield":{"id":"https://openalex.org/subfields/1707","display_name":"Computer Vision and Pattern Recognition"},"field":{"id":"https://openalex.org/fields/17","display_name":"Computer Science"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}}],"keywords":[{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.723013162612915},{"id":"https://openalex.org/keywords/parallelism","display_name":"Parallelism (grammar)","score":0.6346182823181152},{"id":"https://openalex.org/keywords/parallel-computing","display_name":"Parallel computing","score":0.5928351283073425},{"id":"https://openalex.org/keywords/graph","display_name":"Graph","score":0.5452086329460144},{"id":"https://openalex.org/keywords/theoretical-computer-science","display_name":"Theoretical computer science","score":0.2934221029281616}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.723013162612915},{"id":"https://openalex.org/C2781172179","wikidata":"https://www.wikidata.org/wiki/Q853109","display_name":"Parallelism (grammar)","level":2,"score":0.6346182823181152},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.5928351283073425},{"id":"https://openalex.org/C132525143","wikidata":"https://www.wikidata.org/wiki/Q141488","display_name":"Graph","level":2,"score":0.5452086329460144},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.2934221029281616}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.14778/3705829.3705837","is_oa":false,"landing_page_url":"https://doi.org/10.14778/3705829.3705837","pdf_url":null,"source":{"id":"https://openalex.org/S4210226185","display_name":"Proceedings of the VLDB Endowment","issn_l":"2150-8097","issn":["2150-8097"],"is_oa":false,"is_in_doaj":false,"is_core":true,"host_organization":"https://openalex.org/P4310319798","host_organization_name":"Association for Computing Machinery","host_organization_lineage":["https://openalex.org/P4310319798"],"host_organization_lineage_names":["Association for Computing Machinery"],"type":"journal"},"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the VLDB Endowment","raw_type":"journal-article"}],"best_oa_location":null,"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":22,"referenced_works":["https://openalex.org/W36951717","https://openalex.org/W2560674852","https://openalex.org/W2747329762","https://openalex.org/W2807021761","https://openalex.org/W2962946486","https://openalex.org/W2970929262","https://openalex.org/W3007364240","https://openalex.org/W3045200674","https://openalex.org/W3086238199","https://openalex.org/W3096566397","https://openalex.org/W3100848837","https://openalex.org/W3133028082","https://openalex.org/W3152893301","https://openalex.org/W3206504463","https://openalex.org/W4283314525","https://openalex.org/W4285506300","https://openalex.org/W4288070868","https://openalex.org/W4290877962","https://openalex.org/W4381328668","https://openalex.org/W4383749445","https://openalex.org/W4389495241","https://openalex.org/W4392024605"],"related_works":["https://openalex.org/W4391375266","https://openalex.org/W2899084033","https://openalex.org/W2748952813","https://openalex.org/W2390279801","https://openalex.org/W2045183646","https://openalex.org/W2162409446","https://openalex.org/W4391913857","https://openalex.org/W2023505575","https://openalex.org/W1595672120","https://openalex.org/W4230999561"],"abstract_inverted_index":{"Graph":[0],"neural":[1],"networks":[2],"(GNNs)":[3],"have":[4],"emerged":[5],"as":[6],"a":[7,107,124,156,188,198],"promising":[8],"direction.":[9],"Training":[10],"large-scale":[11],"graphs":[12,167],"that":[13,203],"relies":[14],"on":[15,96,197],"distributed":[16,23,51,73,189],"computing":[17],"power":[18],"poses":[19],"new":[20],"challenges.":[21],"Existing":[22],"GNN":[24,74,76,114,190,210],"systems":[25,211],"leverage":[26,69],"data":[27],"parallelism":[28,71,78,116],"by":[29,83,143,176],"partitioning":[30,84],"the":[31,43,47,101,139,163,183],"input":[32],"graph":[33,48,88,134],"and":[34,57,179,215],"distributing":[35],"it":[36],"to":[37,42,106,129,161],"multiple":[38],"workers.":[39],"However,":[40],"due":[41],"irregular":[44],"nature":[45],"of":[46,87,165],"structure,":[49],"existing":[50],"approaches":[52],"suffer":[53],"from":[54,133],"unbalanced":[55],"workloads":[56],"high":[58],"overhead":[59,141],"in":[60],"managing":[61],"cross-worker":[62,80],"vertex":[63,81],"dependencies.":[64],"In":[65],"this":[66],"paper,":[67],"we":[68,122,154,186],"tensor":[70,77,115],"for":[72],"training.":[75],"eliminates":[79],"dependencies":[82],"features":[85],"instead":[86],"structures.":[89],"Different":[90],"workers":[91],"are":[92],"assigned":[93],"training":[94,127,164,191],"tasks":[95],"different":[97],"feature":[98],"slices":[99],"with":[100],"same":[102],"dimensional":[103],"size,":[104],"leading":[105],"complete":[108,151],"load":[109],"balance.":[110],"We":[111],"achieve":[112],"efficient":[113],"through":[117],"two":[118],"critical":[119],"functions.":[120],"Firstly,":[121],"employ":[123,155],"generalized":[125],"decoupled":[126],"framework":[128],"decouple":[130],"NN":[131,144],"operations":[132,145],"aggregation":[135],"operations,":[136],"significantly":[137],"reducing":[138],"communication":[140,178],"caused":[142],"which":[146],"must":[147],"be":[148],"computed":[149],"using":[150],"features.":[152],"Secondly,":[153],"memory-efficient":[157],"task":[158],"scheduling":[159],"strategy":[160],"support":[162],"large":[166],"exceeding":[168],"single":[169],"GPU":[170],"memory,":[171],"while":[172],"further":[173],"improving":[174],"performance":[175],"overlapping":[177],"computation.":[180],"By":[181],"integrating":[182],"above":[184],"techniques,":[185],"propose":[187],"system":[192],"NeutronTP.":[193],"Our":[194],"experimental":[195],"results":[196],"16-node":[199],"Aliyun":[200],"cluster":[201],"demonstrate":[202],"NeutronTP":[204],"achieves":[205],"1.29\u00d7-8.72\u00d7":[206],"speedup":[207],"over":[208],"state-of-the-art":[209],"including":[212],"DistDGL,":[213],"NeutronStar,":[214],"Sancus.":[216]},"counts_by_year":[{"year":2026,"cited_by_count":3},{"year":2025,"cited_by_count":4}],"updated_date":"2026-06-11T09:08:48.828518","created_date":"2025-10-10T00:00:00"}