{"id":"https://openalex.org/W7163047298","doi":"https://doi.org/10.48550/arxiv.2605.31500","title":"On Efficient Scaling of GNNs via IO-Aware Layers Implementations","display_name":"On Efficient Scaling of GNNs via IO-Aware Layers Implementations","publication_year":2026,"publication_date":"2026-05-29","ids":{"openalex":"https://openalex.org/W7163047298","doi":"https://doi.org/10.48550/arxiv.2605.31500"},"language":null,"primary_location":{"id":"doi:10.48550/arxiv.2605.31500","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.31500","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"type":"preprint","indexed_in":["datacite"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://doi.org/10.48550/arxiv.2605.31500","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5101927989","display_name":"Daria Fomina","orcid":"https://orcid.org/0000-0002-6341-7810"},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Fomina, Daria","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137609560","display_name":"Daniil Krasylnikov","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Krasylnikov, Daniil","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5103792159","display_name":"Alexey Boykov","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Boykov, Alexey","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5137560093","display_name":"Andrey Dolgovyazov","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Dolgovyazov, Andrey","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"middle","author":{"id":"https://openalex.org/A5133359448","display_name":"Vyacheslav Zhdanovskiy","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Zhdanovskiy, Vyacheslav","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]},{"author_position":"last","author":{"id":"https://openalex.org/A5114420302","display_name":"Fedor Velikonivtsev","orcid":null},"institutions":[],"countries":[],"is_corresponding":false,"raw_author_name":"Velikonivtsev, Fedor","raw_affiliation_strings":[],"raw_orcid":null,"affiliations":[]}],"institutions":[],"countries_distinct_count":0,"institutions_distinct_count":0,"corresponding_author_ids":[],"corresponding_institution_ids":[],"apc_list":null,"apc_paid":null,"fwci":null,"has_fulltext":false,"cited_by_count":0,"citation_normalized_percentile":null,"cited_by_percentile_year":null,"biblio":{"volume":null,"issue":null,"first_page":null,"last_page":null},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T11273","display_name":"Advanced Graph Neural Networks","score":0.9478999972343445,"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.9478999972343445,"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/T12292","display_name":"Graph Theory and Algorithms","score":0.030700000002980232,"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"}},{"id":"https://openalex.org/T14347","display_name":"Big Data and Digital Economy","score":0.002899999963119626,"subfield":{"id":"https://openalex.org/subfields/1710","display_name":"Information Systems"},"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/speedup","display_name":"Speedup","score":0.9708999991416931},{"id":"https://openalex.org/keywords/scalability","display_name":"Scalability","score":0.725600004196167},{"id":"https://openalex.org/keywords/scaling","display_name":"Scaling","score":0.569100022315979},{"id":"https://openalex.org/keywords/kernel","display_name":"Kernel (algebra)","score":0.5238000154495239},{"id":"https://openalex.org/keywords/limiting","display_name":"Limiting","score":0.4381999969482422},{"id":"https://openalex.org/keywords/implementation","display_name":"Implementation","score":0.4348999857902527},{"id":"https://openalex.org/keywords/graph","display_name":"Graph","score":0.42559999227523804},{"id":"https://openalex.org/keywords/cache","display_name":"Cache","score":0.3628999888896942},{"id":"https://openalex.org/keywords/graph-kernel","display_name":"Graph kernel","score":0.33489999175071716}],"concepts":[{"id":"https://openalex.org/C68339613","wikidata":"https://www.wikidata.org/wiki/Q1549489","display_name":"Speedup","level":2,"score":0.9708999991416931},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7953000068664551},{"id":"https://openalex.org/C48044578","wikidata":"https://www.wikidata.org/wiki/Q727490","display_name":"Scalability","level":2,"score":0.725600004196167},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.6442999839782715},{"id":"https://openalex.org/C99844830","wikidata":"https://www.wikidata.org/wiki/Q102441924","display_name":"Scaling","level":2,"score":0.569100022315979},{"id":"https://openalex.org/C74193536","wikidata":"https://www.wikidata.org/wiki/Q574844","display_name":"Kernel (algebra)","level":2,"score":0.5238000154495239},{"id":"https://openalex.org/C188198153","wikidata":"https://www.wikidata.org/wiki/Q1613840","display_name":"Limiting","level":2,"score":0.4381999969482422},{"id":"https://openalex.org/C26713055","wikidata":"https://www.wikidata.org/wiki/Q245962","display_name":"Implementation","level":2,"score":0.4348999857902527},{"id":"https://openalex.org/C132525143","wikidata":"https://www.wikidata.org/wiki/Q141488","display_name":"Graph","level":2,"score":0.42559999227523804},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.41440001130104065},{"id":"https://openalex.org/C115537543","wikidata":"https://www.wikidata.org/wiki/Q165596","display_name":"Cache","level":2,"score":0.3628999888896942},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.3499000072479248},{"id":"https://openalex.org/C100595998","wikidata":"https://www.wikidata.org/wiki/Q11731931","display_name":"Graph kernel","level":5,"score":0.33489999175071716},{"id":"https://openalex.org/C155281189","wikidata":"https://www.wikidata.org/wiki/Q3518150","display_name":"Tensor (intrinsic definition)","level":2,"score":0.32089999318122864},{"id":"https://openalex.org/C74912251","wikidata":"https://www.wikidata.org/wiki/Q6815727","display_name":"Memory footprint","level":2,"score":0.3197999894618988},{"id":"https://openalex.org/C63479239","wikidata":"https://www.wikidata.org/wiki/Q7353546","display_name":"Robustness (evolution)","level":3,"score":0.3188999891281128},{"id":"https://openalex.org/C189783530","wikidata":"https://www.wikidata.org/wiki/Q352090","display_name":"CPU cache","level":3,"score":0.3125},{"id":"https://openalex.org/C82687282","wikidata":"https://www.wikidata.org/wiki/Q66221","display_name":"Auxiliary memory","level":2,"score":0.298799991607666},{"id":"https://openalex.org/C162319229","wikidata":"https://www.wikidata.org/wiki/Q175263","display_name":"Data structure","level":2,"score":0.2874000072479248},{"id":"https://openalex.org/C91481028","wikidata":"https://www.wikidata.org/wiki/Q1054686","display_name":"Distributed memory","level":3,"score":0.2808000147342682},{"id":"https://openalex.org/C48903430","wikidata":"https://www.wikidata.org/wiki/Q491370","display_name":"Graph partition","level":3,"score":0.26910001039505005},{"id":"https://openalex.org/C111335779","wikidata":"https://www.wikidata.org/wiki/Q3454686","display_name":"Reduction (mathematics)","level":2,"score":0.2614000141620636},{"id":"https://openalex.org/C2781357197","wikidata":"https://www.wikidata.org/wiki/Q5757597","display_name":"High memory","level":2,"score":0.2563000023365021},{"id":"https://openalex.org/C3017489831","wikidata":"https://www.wikidata.org/wiki/Q2393193","display_name":"Running time","level":2,"score":0.2551000118255615},{"id":"https://openalex.org/C106937863","wikidata":"https://www.wikidata.org/wiki/Q7236518","display_name":"Power graph analysis","level":3,"score":0.2513999938964844}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.48550/arxiv.2605.31500","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.31500","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":null,"raw_source_name":null,"raw_type":"Preprint"}],"best_oa_location":{"id":"doi:10.48550/arxiv.2605.31500","is_oa":true,"landing_page_url":"https://doi.org/10.48550/arxiv.2605.31500","pdf_url":null,"source":{"id":"https://openalex.org/S4306400194","display_name":"arXiv (Cornell University)","issn_l":null,"issn":null,"is_oa":true,"is_in_doaj":false,"is_core":false,"host_organization":"https://openalex.org/I205783295","host_organization_name":"Cornell University","host_organization_lineage":["https://openalex.org/I205783295"],"host_organization_lineage_names":[],"type":"repository"},"license":"cc-by","license_id":"https://openalex.org/licenses/cc-by","version":null,"is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"Preprint"},"sustainable_development_goals":[],"awards":[],"funders":[],"has_content":{"grobid_xml":false,"pdf":false},"content_urls":null,"referenced_works_count":0,"referenced_works":[],"related_works":[],"abstract_inverted_index":{"Graph":[0,121],"Neural":[1],"Networks":[2],"(GNNs)":[3],"are":[4],"bottlenecked":[5],"by":[6,151],"sparse,":[7],"irregular":[8],"memory":[9,31,150],"access.":[10],"Popular":[11],"frameworks":[12],"such":[13],"as":[14,195],"DGL":[15,180],"and":[16,33,43,46,61,79,90,181],"PyTorch":[17],"Geometric":[18],"support":[19,199],"general":[20],"message":[21],"passing,":[22],"but":[23],"complex":[24],"layers":[25,51,63],"often":[26],"materialize":[27],"edge-wise":[28],"intermediates,":[29],"increasing":[30],"traffic":[32],"limiting":[34],"scalability":[35],"on":[36,96,133],"large":[37],"graphs.":[38,84],"We":[39,85,191],"take":[40],"an":[41],"I/O-":[42],"arithmetic-intensity--centric":[44],"view":[45],"show":[47],"that":[48,73,92],"widely":[49],"used":[50],"fall":[52],"into":[53],"three":[54],"kernel":[55,98],"families:":[56],"SpMM-based":[57,169],"convolutions,":[58],"reduction-based":[59],"aggregations,":[60],"attention-based":[62],"(GATv2/Graph":[64],"Transformer).":[65],"For":[66,168],"each":[67],"family,":[68],"we":[69,139],"develop":[70],"GPU":[71],"kernels":[72,104,114,160],"reduce":[74],"data":[75],"movement,":[76],"improve":[77],"locality,":[78],"remain":[80],"robust":[81],"across":[82],"realistic":[83],"also":[86],"study":[87],"graph":[88],"reordering":[89],"find":[91],"its":[93],"impact":[94],"depends":[95],"the":[97,187],"mapping:":[99],"it":[100],"benefits":[101],"neighbor-parallel":[102],"(gather-dominated)":[103],"more":[105],"consistently":[106],"than":[107],"feature-parallel":[108],"designs.":[109],"Empirically,":[110],"our":[111,193],"fused":[112],"attention":[113],"reach":[115,140],"up":[116,130,141,152,162,175],"to":[117,131,142,153,163,176,198],"$\\textbf{3.9}\\times$":[118],"speedup":[119,144,165,178],"for":[120,137],"Transformer":[122],"(median":[123,145,155,166],"$\\textbf{1.6}\\times$),":[124],"with":[125],"Tensor":[126],"Core":[127],"(block-sparse)":[128],"variants":[129],"$\\textbf{7.3}\\times$":[132],"locally":[134],"dense":[135],"graphs;":[136],"GATv2":[138],"$\\textbf{8.5}\\times$":[143],"$\\textbf{2.0}\\times$)":[146],"while":[147],"reducing":[148],"peak":[149],"$\\textbf{76}\\times$":[154],"$\\textbf{6}\\times$).":[156],"Our":[157],"degree-aware":[158],"reduction":[159],"achieve":[161],"$\\textbf{10}\\times$":[164],"$\\textbf{2.6}\\times$).":[167],"layers,":[170],"properly":[171],"cached":[172],"cuSPARSE":[173],"achieves":[174],"$\\textbf{8}\\times$":[177],"over":[179],"outperforms":[182],"evaluated":[183],"custom":[184],"baselines":[185],"in":[186],"majority":[188],"of":[189],"evaluations.":[190],"release":[192],"implementations":[194],"drop-in":[196],"replacements":[197],"reproducible,":[200],"hardware-aware":[201],"GNN":[202],"acceleration.":[203]},"counts_by_year":[],"updated_date":"2026-07-01T06:00:48.157686","created_date":"2026-06-02T00:00:00"}