{"id":"https://openalex.org/W4296604568","doi":"https://doi.org/10.1145/3523227.3546760","title":"TinyKG: Memory-Efficient Training Framework for Knowledge Graph Neural Recommender Systems","display_name":"TinyKG: Memory-Efficient Training Framework for Knowledge Graph Neural Recommender Systems","publication_year":2022,"publication_date":"2022-09-13","ids":{"openalex":"https://openalex.org/W4296604568","doi":"https://doi.org/10.1145/3523227.3546760"},"language":"en","primary_location":{"id":"doi:10.1145/3523227.3546760","is_oa":false,"landing_page_url":"https://doi.org/10.1145/3523227.3546760","pdf_url":null,"source":null,"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 16th ACM Conference on Recommender Systems","raw_type":"proceedings-article"},"type":"preprint","indexed_in":["arxiv","crossref"],"open_access":{"is_oa":true,"oa_status":"green","oa_url":"https://arxiv.org/pdf/2212.04540","any_repository_has_fulltext":true},"authorships":[{"author_position":"first","author":{"id":"https://openalex.org/A5004341287","display_name":"Huiyuan Chen","orcid":"https://orcid.org/0000-0002-6360-558X"},"institutions":[{"id":"https://openalex.org/I4210148469","display_name":"Visa (United States)","ror":"https://ror.org/05t1y0b59","country_code":"US","type":"company","lineage":["https://openalex.org/I4210148469"]}],"countries":["US"],"is_corresponding":true,"raw_author_name":"Huiyuan Chen","raw_affiliation_strings":["Visa Research, United States"],"affiliations":[{"raw_affiliation_string":"Visa Research, United States","institution_ids":["https://openalex.org/I4210148469"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5100621457","display_name":"Xiaoting Li","orcid":"https://orcid.org/0000-0002-1538-3644"},"institutions":[{"id":"https://openalex.org/I4210148469","display_name":"Visa (United States)","ror":"https://ror.org/05t1y0b59","country_code":"US","type":"company","lineage":["https://openalex.org/I4210148469"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Xiaoting Li","raw_affiliation_strings":["Visa Research, United States"],"affiliations":[{"raw_affiliation_string":"Visa Research, United States","institution_ids":["https://openalex.org/I4210148469"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5071607114","display_name":"Kaixiong Zhou","orcid":"https://orcid.org/0000-0001-5226-8736"},"institutions":[{"id":"https://openalex.org/I74775410","display_name":"Rice University","ror":"https://ror.org/008zs3103","country_code":"US","type":"education","lineage":["https://openalex.org/I74775410"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Kaixiong Zhou","raw_affiliation_strings":["Rice University, United States"],"affiliations":[{"raw_affiliation_string":"Rice University, United States","institution_ids":["https://openalex.org/I74775410"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5068477431","display_name":"Xia Hu","orcid":"https://orcid.org/0000-0003-2234-3226"},"institutions":[{"id":"https://openalex.org/I74775410","display_name":"Rice University","ror":"https://ror.org/008zs3103","country_code":"US","type":"education","lineage":["https://openalex.org/I74775410"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Xia Hu","raw_affiliation_strings":["Rice University, United States"],"affiliations":[{"raw_affiliation_string":"Rice University, United States","institution_ids":["https://openalex.org/I74775410"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5011045620","display_name":"Chin\u2010Chia Michael Yeh","orcid":"https://orcid.org/0000-0002-9807-2963"},"institutions":[{"id":"https://openalex.org/I4210148469","display_name":"Visa (United States)","ror":"https://ror.org/05t1y0b59","country_code":"US","type":"company","lineage":["https://openalex.org/I4210148469"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Chin-Chia Michael Yeh","raw_affiliation_strings":["Visa Research, United States"],"affiliations":[{"raw_affiliation_string":"Visa Research, United States","institution_ids":["https://openalex.org/I4210148469"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5101563636","display_name":"Yan Zheng","orcid":"https://orcid.org/0000-0003-4422-9889"},"institutions":[{"id":"https://openalex.org/I4210148469","display_name":"Visa (United States)","ror":"https://ror.org/05t1y0b59","country_code":"US","type":"company","lineage":["https://openalex.org/I4210148469"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Yan Zheng","raw_affiliation_strings":["Visa Research, United States"],"affiliations":[{"raw_affiliation_string":"Visa Research, United States","institution_ids":["https://openalex.org/I4210148469"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5075211218","display_name":"Hao Yang","orcid":"https://orcid.org/0000-0003-3020-9828"},"institutions":[{"id":"https://openalex.org/I4210148469","display_name":"Visa (United States)","ror":"https://ror.org/05t1y0b59","country_code":"US","type":"company","lineage":["https://openalex.org/I4210148469"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Hao Yang","raw_affiliation_strings":["Visa Research, United States"],"affiliations":[{"raw_affiliation_string":"Visa Research, United States","institution_ids":["https://openalex.org/I4210148469"]}]}],"institutions":[],"countries_distinct_count":1,"institutions_distinct_count":7,"corresponding_author_ids":["https://openalex.org/A5004341287"],"corresponding_institution_ids":["https://openalex.org/I4210148469"],"apc_list":null,"apc_paid":null,"fwci":0.966,"has_fulltext":false,"cited_by_count":7,"citation_normalized_percentile":{"value":0.79714434,"is_in_top_1_percent":false,"is_in_top_10_percent":false},"cited_by_percentile_year":{"min":94,"max":98},"biblio":{"volume":null,"issue":null,"first_page":"257","last_page":"267"},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T11273","display_name":"Advanced Graph Neural Networks","score":0.9998999834060669,"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.9998999834060669,"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/T10203","display_name":"Recommender Systems and Techniques","score":0.9993000030517578,"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"}},{"id":"https://openalex.org/T11612","display_name":"Stochastic Gradient Optimization Techniques","score":0.9908000230789185,"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"}}],"keywords":[{"id":"https://openalex.org/keywords/memory-footprint","display_name":"Memory footprint","score":0.869644284248352},{"id":"https://openalex.org/keywords/computer-science","display_name":"Computer science","score":0.8663631677627563},{"id":"https://openalex.org/keywords/quantization","display_name":"Quantization (signal processing)","score":0.7528514266014099},{"id":"https://openalex.org/keywords/cache","display_name":"Cache","score":0.5209394693374634},{"id":"https://openalex.org/keywords/graph","display_name":"Graph","score":0.49616989493370056},{"id":"https://openalex.org/keywords/footprint","display_name":"Footprint","score":0.4647964835166931},{"id":"https://openalex.org/keywords/artificial-neural-network","display_name":"Artificial neural network","score":0.45703306794166565},{"id":"https://openalex.org/keywords/knowledge-graph","display_name":"Knowledge graph","score":0.45144057273864746},{"id":"https://openalex.org/keywords/computer-engineering","display_name":"Computer engineering","score":0.44729265570640564},{"id":"https://openalex.org/keywords/parallel-computing","display_name":"Parallel computing","score":0.4167254865169525},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.3802434802055359},{"id":"https://openalex.org/keywords/machine-learning","display_name":"Machine learning","score":0.3800659775733948},{"id":"https://openalex.org/keywords/theoretical-computer-science","display_name":"Theoretical computer science","score":0.34933003783226013},{"id":"https://openalex.org/keywords/algorithm","display_name":"Algorithm","score":0.2289028763771057},{"id":"https://openalex.org/keywords/programming-language","display_name":"Programming language","score":0.0817224383354187}],"concepts":[{"id":"https://openalex.org/C74912251","wikidata":"https://www.wikidata.org/wiki/Q6815727","display_name":"Memory footprint","level":2,"score":0.869644284248352},{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.8663631677627563},{"id":"https://openalex.org/C28855332","wikidata":"https://www.wikidata.org/wiki/Q198099","display_name":"Quantization (signal processing)","level":2,"score":0.7528514266014099},{"id":"https://openalex.org/C115537543","wikidata":"https://www.wikidata.org/wiki/Q165596","display_name":"Cache","level":2,"score":0.5209394693374634},{"id":"https://openalex.org/C132525143","wikidata":"https://www.wikidata.org/wiki/Q141488","display_name":"Graph","level":2,"score":0.49616989493370056},{"id":"https://openalex.org/C132943942","wikidata":"https://www.wikidata.org/wiki/Q2562511","display_name":"Footprint","level":2,"score":0.4647964835166931},{"id":"https://openalex.org/C50644808","wikidata":"https://www.wikidata.org/wiki/Q192776","display_name":"Artificial neural network","level":2,"score":0.45703306794166565},{"id":"https://openalex.org/C2987255567","wikidata":"https://www.wikidata.org/wiki/Q33002955","display_name":"Knowledge graph","level":2,"score":0.45144057273864746},{"id":"https://openalex.org/C113775141","wikidata":"https://www.wikidata.org/wiki/Q428691","display_name":"Computer engineering","level":1,"score":0.44729265570640564},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.4167254865169525},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.3802434802055359},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.3800659775733948},{"id":"https://openalex.org/C80444323","wikidata":"https://www.wikidata.org/wiki/Q2878974","display_name":"Theoretical computer science","level":1,"score":0.34933003783226013},{"id":"https://openalex.org/C11413529","wikidata":"https://www.wikidata.org/wiki/Q8366","display_name":"Algorithm","level":1,"score":0.2289028763771057},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.0817224383354187},{"id":"https://openalex.org/C86803240","wikidata":"https://www.wikidata.org/wiki/Q420","display_name":"Biology","level":0,"score":0.0},{"id":"https://openalex.org/C151730666","wikidata":"https://www.wikidata.org/wiki/Q7205","display_name":"Paleontology","level":1,"score":0.0}],"mesh":[],"locations_count":2,"locations":[{"id":"doi:10.1145/3523227.3546760","is_oa":false,"landing_page_url":"https://doi.org/10.1145/3523227.3546760","pdf_url":null,"source":null,"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Proceedings of the 16th ACM Conference on Recommender Systems","raw_type":"proceedings-article"},{"id":"pmh:oai:arXiv.org:2212.04540","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2212.04540","pdf_url":"https://arxiv.org/pdf/2212.04540","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":null,"license_id":null,"version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"}],"best_oa_location":{"id":"pmh:oai:arXiv.org:2212.04540","is_oa":true,"landing_page_url":"http://arxiv.org/abs/2212.04540","pdf_url":"https://arxiv.org/pdf/2212.04540","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":null,"license_id":null,"version":"submittedVersion","is_accepted":false,"is_published":false,"raw_source_name":null,"raw_type":"text"},"sustainable_development_goals":[{"score":0.5899999737739563,"id":"https://metadata.un.org/sdg/9","display_name":"Industry, innovation and infrastructure"}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":24,"referenced_works":["https://openalex.org/W2509893387","https://openalex.org/W2893775232","https://openalex.org/W2911778742","https://openalex.org/W2912516411","https://openalex.org/W2926442184","https://openalex.org/W2945623882","https://openalex.org/W2963911286","https://openalex.org/W3012479151","https://openalex.org/W3023393331","https://openalex.org/W3034364571","https://openalex.org/W3035109558","https://openalex.org/W3035403290","https://openalex.org/W3095602948","https://openalex.org/W3099403815","https://openalex.org/W3129482887","https://openalex.org/W3155919942","https://openalex.org/W3156833186","https://openalex.org/W3200593175","https://openalex.org/W4224320470","https://openalex.org/W4281861579","https://openalex.org/W4288057195","https://openalex.org/W4290943921","https://openalex.org/W4290948420","https://openalex.org/W6784694379"],"related_works":["https://openalex.org/W2330537534","https://openalex.org/W2368534554","https://openalex.org/W2148469257","https://openalex.org/W2999244305","https://openalex.org/W2168827721","https://openalex.org/W2152099439","https://openalex.org/W3137434606","https://openalex.org/W4372263373","https://openalex.org/W2929170389","https://openalex.org/W2950268673"],"abstract_inverted_index":{"There":[0],"has":[1],"been":[2],"an":[3],"explosion":[4],"of":[5,34,133,149,202,215,238],"interest":[6],"in":[7,74,79,112,140,151,168,248],"designing":[8],"various":[9],"Knowledge":[10],"Graph":[11],"Neural":[12],"Networks":[13],"(KGNNs),":[14],"which":[15,83,190],"achieve":[16],"state-of-the-art":[17],"performance":[18,27,214],"and":[19],"provide":[20],"great":[21],"explainability":[22],"for":[23,116,128,130],"recommendation.":[24,134],"The":[25],"promising":[26],"is":[28,48,204],"mainly":[29],"resulting":[30],"from":[31],"their":[32],"capability":[33],"capturing":[35],"high-order":[36],"proximity":[37],"messages":[38],"over":[39],"the":[40,52,57,60,80,131,141,152,156,175,188,198,213,235],"knowledge":[41],"graphs.":[42,118],"However,":[43],"training":[44,126,199],"KGNNs":[45,111,129,203,254],"at":[46],"scale":[47],"challenging":[49],"due":[50],"to":[51,68,76,85,109,165,170,186,252],"high":[53],"memory":[54,89,200,236],"usage.":[55],"In":[56],"forward":[58,142],"pass,":[59,82,158],"automatic":[61],"differentiation":[62],"engines":[63],"(\\textsl{e.g.},":[64],"TensorFlow/PyTorch)":[65],"generally":[66],"need":[67],"cache":[69],"all":[70],"intermediate":[71],"activation":[72,239],"maps":[73,240],"order":[75,169],"compute":[77,171],"gradients":[78],"backward":[81,157],"leads":[84],"a":[86,104,123,146,180],"large":[87],"GPU":[88,153],"footprint.":[90],"Existing":[91],"work":[92],"solves":[93],"this":[94,102],"problem":[95],"by":[96],"utilizing":[97],"multi-GPU":[98],"distributed":[99],"frameworks.":[100],"Nonetheless,":[101],"poses":[103],"practical":[105],"challenge":[106],"when":[107],"seeking":[108],"deploy":[110,253],"memory-constrained":[113,256],"environments,":[114],"especially":[115],"industry-scale":[117],"Here":[119],"we":[120,218],"present":[121],"TinyKG,":[122,217],"memory-efficient":[124],"GPU-based":[125],"framework":[127],"tasks":[132],"Specifically,":[135],"TinyKG":[136,178,229],"uses":[137],"exact":[138],"activations":[139,150,161],"pass":[143],"while":[144],"storing":[145],"quantized":[147],"version":[148],"buffers.":[154],"During":[155],"these":[159],"low-precision":[160],"are":[162],"dequantized":[163],"back":[164],"full-precision":[166],"tensors,":[167],"gradients.":[172],"To":[173,211],"reduce":[174],"quantization":[176,184,232],"errors,":[177],"applies":[179],"simple":[181],"yet":[182],"effective":[183],"algorithm":[185],"compress":[187],"activations,":[189],"ensures":[191],"unbiasedness":[192],"with":[193,207,230,241,245],"low":[194],"variance.":[195],"As":[196],"such,":[197],"footprint":[201,237],"largely":[205],"reduced":[206],"negligible":[208],"accuracy":[209],"loss.":[210],"evaluate":[212],"our":[216,228],"conduct":[219],"comprehensive":[220],"experiments":[221],"on":[222,255],"real-world":[223],"datasets.":[224],"We":[225],"found":[226],"that":[227],"INT2":[231],"aggressively":[233],"reduces":[234],"$7":[242],"\\times$,":[243],"only":[244],"$2\\%$":[246],"loss":[247],"accuracy,":[249],"allowing":[250],"us":[251],"devices.":[257]},"counts_by_year":[{"year":2024,"cited_by_count":2},{"year":2023,"cited_by_count":5}],"updated_date":"2026-03-20T23:20:44.827607","created_date":"2025-10-10T00:00:00"}