{"id":"https://openalex.org/W3035701850","doi":"https://doi.org/10.1080/10618600.2022.2067548","title":"Survival Regression with Accelerated Failure Time Model in XGBoost","display_name":"Survival Regression with Accelerated Failure Time Model in XGBoost","publication_year":2022,"publication_date":"2022-04-25","ids":{"openalex":"https://openalex.org/W3035701850","doi":"https://doi.org/10.1080/10618600.2022.2067548","mag":"3035701850"},"language":"en","primary_location":{"id":"doi:10.1080/10618600.2022.2067548","is_oa":false,"landing_page_url":"https://doi.org/10.1080/10618600.2022.2067548","pdf_url":null,"source":{"id":"https://openalex.org/S76159266","display_name":"Journal of Computational and Graphical Statistics","issn_l":"1061-8600","issn":["1061-8600","1537-2715"],"is_oa":false,"is_in_doaj":false,"is_core":true,"host_organization":"https://openalex.org/P4310320547","host_organization_name":"Taylor & Francis","host_organization_lineage":["https://openalex.org/P4310320547"],"host_organization_lineage_names":["Taylor & Francis"],"type":"journal"},"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Journal of Computational and Graphical Statistics","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/A5066057007","display_name":"Avinash Barnwal","orcid":null},"institutions":[{"id":"https://openalex.org/I59553526","display_name":"Stony Brook University","ror":"https://ror.org/05qghxh33","country_code":"US","type":"education","lineage":["https://openalex.org/I59553526"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Avinash Barnwal","raw_affiliation_strings":["Department of Statistics, Stony Brook University, Stony Brook, NY"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"Department of Statistics, Stony Brook University, Stony Brook, NY","institution_ids":["https://openalex.org/I59553526"]}]},{"author_position":"middle","author":{"id":"https://openalex.org/A5030047934","display_name":"Hyunsu Cho","orcid":"https://orcid.org/0000-0003-0182-6376"},"institutions":[{"id":"https://openalex.org/I4210127875","display_name":"Nvidia (United States)","ror":"https://ror.org/03jdj4y14","country_code":"US","type":"company","lineage":["https://openalex.org/I4210127875"]}],"countries":["US"],"is_corresponding":true,"raw_author_name":"Hyunsu Cho","raw_affiliation_strings":["NVIDIA, Santa Clara, CA"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"NVIDIA, Santa Clara, CA","institution_ids":["https://openalex.org/I4210127875"]}]},{"author_position":"last","author":{"id":"https://openalex.org/A5035679840","display_name":"Toby Dylan Hocking","orcid":"https://orcid.org/0000-0002-3146-0865"},"institutions":[{"id":"https://openalex.org/I203172682","display_name":"Northern Arizona University","ror":"https://ror.org/0272j5188","country_code":"US","type":"education","lineage":["https://openalex.org/I203172682"]}],"countries":["US"],"is_corresponding":false,"raw_author_name":"Toby Hocking","raw_affiliation_strings":["School of Informatics, Computing, and Cyber Systems, Northern Arizona University, Flagstaff, AZ"],"raw_orcid":null,"affiliations":[{"raw_affiliation_string":"School of Informatics, Computing, and Cyber Systems, Northern Arizona University, Flagstaff, AZ","institution_ids":["https://openalex.org/I203172682"]}]}],"institutions":[],"countries_distinct_count":1,"institutions_distinct_count":3,"corresponding_author_ids":["https://openalex.org/A5030047934"],"corresponding_institution_ids":["https://openalex.org/I4210127875"],"apc_list":null,"apc_paid":null,"fwci":10.3646,"has_fulltext":false,"cited_by_count":92,"citation_normalized_percentile":{"value":0.98523682,"is_in_top_1_percent":false,"is_in_top_10_percent":true},"cited_by_percentile_year":{"min":95,"max":100},"biblio":{"volume":"31","issue":"4","first_page":"1292","last_page":"1302"},"is_retracted":false,"is_paratext":false,"is_xpac":false,"primary_topic":{"id":"https://openalex.org/T12535","display_name":"Machine Learning and Data Classification","score":0.9980999827384949,"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/T12535","display_name":"Machine Learning and Data Classification","score":0.9980999827384949,"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/T10136","display_name":"Statistical Methods and Inference","score":0.9965999722480774,"subfield":{"id":"https://openalex.org/subfields/2613","display_name":"Statistics and Probability"},"field":{"id":"https://openalex.org/fields/26","display_name":"Mathematics"},"domain":{"id":"https://openalex.org/domains/3","display_name":"Physical Sciences"}},{"id":"https://openalex.org/T10538","display_name":"Data Mining Algorithms and Applications","score":0.989300012588501,"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/computer-science","display_name":"Computer science","score":0.7381834983825684},{"id":"https://openalex.org/keywords/censoring","display_name":"Censoring (clinical trials)","score":0.6991074085235596},{"id":"https://openalex.org/keywords/machine-learning","display_name":"Machine learning","score":0.5531474947929382},{"id":"https://openalex.org/keywords/decision-tree","display_name":"Decision tree","score":0.5529313087463379},{"id":"https://openalex.org/keywords/speedup","display_name":"Speedup","score":0.5251752138137817},{"id":"https://openalex.org/keywords/artificial-intelligence","display_name":"Artificial intelligence","score":0.4636472463607788},{"id":"https://openalex.org/keywords/feature-engineering","display_name":"Feature engineering","score":0.43706265091896057},{"id":"https://openalex.org/keywords/implementation","display_name":"Implementation","score":0.4165080785751343},{"id":"https://openalex.org/keywords/ensemble-learning","display_name":"Ensemble learning","score":0.4122156500816345},{"id":"https://openalex.org/keywords/data-mining","display_name":"Data mining","score":0.35279232263565063},{"id":"https://openalex.org/keywords/statistics","display_name":"Statistics","score":0.17575860023498535},{"id":"https://openalex.org/keywords/deep-learning","display_name":"Deep learning","score":0.16019582748413086},{"id":"https://openalex.org/keywords/parallel-computing","display_name":"Parallel computing","score":0.15549013018608093},{"id":"https://openalex.org/keywords/mathematics","display_name":"Mathematics","score":0.10801711678504944}],"concepts":[{"id":"https://openalex.org/C41008148","wikidata":"https://www.wikidata.org/wiki/Q21198","display_name":"Computer science","level":0,"score":0.7381834983825684},{"id":"https://openalex.org/C137668524","wikidata":"https://www.wikidata.org/wiki/Q189813","display_name":"Censoring (clinical trials)","level":2,"score":0.6991074085235596},{"id":"https://openalex.org/C119857082","wikidata":"https://www.wikidata.org/wiki/Q2539","display_name":"Machine learning","level":1,"score":0.5531474947929382},{"id":"https://openalex.org/C84525736","wikidata":"https://www.wikidata.org/wiki/Q831366","display_name":"Decision tree","level":2,"score":0.5529313087463379},{"id":"https://openalex.org/C68339613","wikidata":"https://www.wikidata.org/wiki/Q1549489","display_name":"Speedup","level":2,"score":0.5251752138137817},{"id":"https://openalex.org/C154945302","wikidata":"https://www.wikidata.org/wiki/Q11660","display_name":"Artificial intelligence","level":1,"score":0.4636472463607788},{"id":"https://openalex.org/C2778827112","wikidata":"https://www.wikidata.org/wiki/Q22245680","display_name":"Feature engineering","level":3,"score":0.43706265091896057},{"id":"https://openalex.org/C26713055","wikidata":"https://www.wikidata.org/wiki/Q245962","display_name":"Implementation","level":2,"score":0.4165080785751343},{"id":"https://openalex.org/C45942800","wikidata":"https://www.wikidata.org/wiki/Q245652","display_name":"Ensemble learning","level":2,"score":0.4122156500816345},{"id":"https://openalex.org/C124101348","wikidata":"https://www.wikidata.org/wiki/Q172491","display_name":"Data mining","level":1,"score":0.35279232263565063},{"id":"https://openalex.org/C105795698","wikidata":"https://www.wikidata.org/wiki/Q12483","display_name":"Statistics","level":1,"score":0.17575860023498535},{"id":"https://openalex.org/C108583219","wikidata":"https://www.wikidata.org/wiki/Q197536","display_name":"Deep learning","level":2,"score":0.16019582748413086},{"id":"https://openalex.org/C173608175","wikidata":"https://www.wikidata.org/wiki/Q232661","display_name":"Parallel computing","level":1,"score":0.15549013018608093},{"id":"https://openalex.org/C33923547","wikidata":"https://www.wikidata.org/wiki/Q395","display_name":"Mathematics","level":0,"score":0.10801711678504944},{"id":"https://openalex.org/C199360897","wikidata":"https://www.wikidata.org/wiki/Q9143","display_name":"Programming language","level":1,"score":0.0}],"mesh":[],"locations_count":1,"locations":[{"id":"doi:10.1080/10618600.2022.2067548","is_oa":false,"landing_page_url":"https://doi.org/10.1080/10618600.2022.2067548","pdf_url":null,"source":{"id":"https://openalex.org/S76159266","display_name":"Journal of Computational and Graphical Statistics","issn_l":"1061-8600","issn":["1061-8600","1537-2715"],"is_oa":false,"is_in_doaj":false,"is_core":true,"host_organization":"https://openalex.org/P4310320547","host_organization_name":"Taylor & Francis","host_organization_lineage":["https://openalex.org/P4310320547"],"host_organization_lineage_names":["Taylor & Francis"],"type":"journal"},"license":null,"license_id":null,"version":"publishedVersion","is_accepted":true,"is_published":true,"raw_source_name":"Journal of Computational and Graphical Statistics","raw_type":"journal-article"}],"best_oa_location":null,"sustainable_development_goals":[{"score":0.6100000143051147,"display_name":"Peace, Justice and strong institutions","id":"https://metadata.un.org/sdg/16"}],"awards":[],"funders":[],"has_content":{"pdf":false,"grobid_xml":false},"content_urls":null,"referenced_works_count":53,"referenced_works":["https://openalex.org/W1506806321","https://openalex.org/W1571132164","https://openalex.org/W1580788756","https://openalex.org/W1594031697","https://openalex.org/W1678356000","https://openalex.org/W1972476693","https://openalex.org/W1979300931","https://openalex.org/W1999393241","https://openalex.org/W2005755239","https://openalex.org/W2017008479","https://openalex.org/W2018901103","https://openalex.org/W2024046085","https://openalex.org/W2027401669","https://openalex.org/W2039656279","https://openalex.org/W2052825782","https://openalex.org/W2059293100","https://openalex.org/W2083972068","https://openalex.org/W2084139018","https://openalex.org/W2086934780","https://openalex.org/W2099620796","https://openalex.org/W2101234009","https://openalex.org/W2106916900","https://openalex.org/W2112740000","https://openalex.org/W2115584760","https://openalex.org/W2120579447","https://openalex.org/W2168871277","https://openalex.org/W2169753597","https://openalex.org/W2295598076","https://openalex.org/W2305849921","https://openalex.org/W2538873949","https://openalex.org/W2604808181","https://openalex.org/W2612690371","https://openalex.org/W2768348081","https://openalex.org/W2787070805","https://openalex.org/W2794014338","https://openalex.org/W2949676527","https://openalex.org/W2963232127","https://openalex.org/W2963896648","https://openalex.org/W2964022491","https://openalex.org/W2973382503","https://openalex.org/W2996717911","https://openalex.org/W3027987014","https://openalex.org/W3092436046","https://openalex.org/W3097349486","https://openalex.org/W3099478002","https://openalex.org/W3099723433","https://openalex.org/W3102476541","https://openalex.org/W3122807153","https://openalex.org/W3147894994","https://openalex.org/W3176867960","https://openalex.org/W4212774754","https://openalex.org/W4246734090","https://openalex.org/W4293241248"],"related_works":["https://openalex.org/W2058965144","https://openalex.org/W2164382479","https://openalex.org/W98480971","https://openalex.org/W2150291671","https://openalex.org/W2027972911","https://openalex.org/W2146343568","https://openalex.org/W2013643406","https://openalex.org/W3136871737","https://openalex.org/W2558685994","https://openalex.org/W4390860632"],"abstract_inverted_index":{"Survival":[0],"regression":[1],"is":[2,14,149],"used":[3],"to":[4,84,112,137],"estimate":[5],"the":[6,86,104,130,150,157,165,168,173,187,198],"relation":[7],"between":[8],"time-to-event":[9],"and":[10,13,25,43,101,122,191],"feature":[11],"variables,":[12],"important":[15],"in":[16,36,49,82,108,117,135,180,186],"application":[17],"domains":[18],"such":[19,38],"as":[20,34,39],"medicine,":[21],"marketing,":[22],"risk":[23],"management,":[24],"sales":[26],"management.":[27],"Nonlinear":[28],"tree":[29],"based":[30],"machine":[31],"learning":[32,76],"algorithms":[33],"implemented":[35],"libraries":[37],"XGBoost,":[40,83],"scikit-learn,":[41],"LightGBM,":[42],"CatBoost":[44],"are":[45,207],"often":[46],"more":[47],"accurate":[48],"practice":[50],"than":[51],"linear":[52],"models.":[53],"However,":[54],"existing":[55],"state-of-the-art":[56],"implementations":[57],"of":[58,94,106,115,129,153,160,178],"tree-based":[59],"models":[60,81],"have":[61],"offered":[62],"limited":[63],"support":[64,87,131],"for":[65,75,88,91,132,204],"survival":[66,89],"regression.":[67],"In":[68],"this":[69,205],"work,":[70],"we":[71,126],"implement":[72],"loss":[73],"functions":[74],"accelerated":[77],"failure":[78],"time":[79],"(AFT)":[80],"increase":[85],"modeling":[90],"different":[92],"kinds":[93],"label":[95],"censoring.":[96],"We":[97],"demonstrate":[98],"with":[99,110],"real":[100],"simulated":[102],"experiments":[103],"effectiveness":[105],"AFT":[107,154,174,179,200],"XGBoost":[109,136,169,181,199],"respect":[111],"a":[113,192],"number":[114],"baselines,":[116],"two":[118],"respects:":[119],"generalization":[120],"performance":[121],"training":[123],"speed.":[124],"Furthermore,":[125],"take":[127],"advantage":[128],"NVIDIA":[133,161],"GPUs":[134],"achieve":[138],"substantial":[139],"speedup":[140],"over":[141],"multi-core":[142],"CPUs.":[143],"To":[144],"our":[145,147],"knowledge,":[146],"work":[148],"first":[151],"implementation":[152],"that":[155],"uses":[156],"processing":[158],"power":[159],"GPUs.":[162],"Starting":[163],"from":[164],"1.2.0":[166],"release,":[167],"package":[170],"natively":[171],"supports":[172],"model.":[175,201],"The":[176],"addition":[177],"has":[182],"had":[183],"significant":[184],"impact":[185],"open":[188],"source":[189],"community,":[190],"few":[193],"statistics":[194],"packages":[195],"now":[196],"use":[197],"Supplementary":[202],"materials":[203],"article":[206],"available":[208],"online.":[209]},"counts_by_year":[{"year":2026,"cited_by_count":14},{"year":2025,"cited_by_count":30},{"year":2024,"cited_by_count":26},{"year":2023,"cited_by_count":13},{"year":2022,"cited_by_count":6},{"year":2021,"cited_by_count":3}],"updated_date":"2026-06-30T13:55:48.251075","created_date":"2025-10-10T00:00:00"}