Change search
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
SurvLoss: A New Survival Loss Function for Neural Networks to Process Censored Data
Center for Applied Intelligent Systems Research (CAISR), Halmstad University.
Stockholm University, Faculty of Social Sciences, Department of Computer and Systems Sciences.
Number of Authors: 22024 (English)In: Proceedings of the European Conference of the PHM Society 2024 / [ed] Phuc Do; Cordelia Ezhilarasu, Prognostics and Health Management Society , 2024, Vol. 8, article id 4052Conference paper, Published paper (Refereed)
Abstract [en]

This paper presents SurvLoss, a novel asymmetric partial loss and error calculation function for survival analysis and regression, enabling the inclusion of censored samples. An observation in a dataset for which the complete information regarding an event of interest is not available is called censored. Censored samples are ubiquitous in the industry and play a crucial role in Prognostics and Health Management (PHM) by providing a realistic representation of data, improving the accuracy of analyses, and supporting better decision-making in various industries and the healthcare sector. The proposed approach can effectively equip the conventional regression loss functions such as Mean Absolute Error (MAE), Mean Squared Error (MSE), or Root Mean Squared Error (RMSE) with the ability to process censored samples. This can impact the field hugely by providing a more accessible usage of neural network models in survival analysis. The proposed survival loss incorporates censored samples by penalizing predictions outside the censoring region and skipping them otherwise. Then, it uses weighted averaging to aggregate the loss from censored samples with the loss from event samples.

Unlike many other methods in the field, the proposed model distinguishes itself by avoiding superficial assumptions and exclusively relies on the available information, considering the entirety of the data.

We compared the proposed loss function with its baseline on two publicly available datasets. The first dataset, called C-MAPSS, is from NASA Turbofan Jet Engines simulation, and the second is a recently published real-world dataset from

SCANIA trucks. The goal of both datasets is to predict the remaining useful life (RUL) of the machines. The experimental results show that optimization algorithms for training deep neural networks like Adam can effectively utilize the proposed loss function to calculate gradients, update the model’s weights, and reduce training and test errors. Moreover, the

proposed model outperformed the baseline by taking advantage of the censored samples. The proposed loss function paves the way for the employment of advanced architectures of neural networks with bigger training sizes in survival analysis.

Place, publisher, year, edition, pages
Prognostics and Health Management Society , 2024. Vol. 8, article id 4052
Series
Proceedings of the European Conference of the Prognostics and Health Management Society (PHME), E-ISSN 2325-016X ; VOL. 8 NO. 1
Keywords [en]
Survival analysis, Regression, Remaining Useful Life estimation, Predictive Maintenance, Loss function, Time series
National Category
Computer Sciences
Research subject
Computer and Systems Sciences
Identifiers
URN: urn:nbn:se:su:diva-232106DOI: 10.36001/phme.2024.v8i1.4052ISBN: 978-1-936263-40-0 (print)OAI: oai:DiVA.org:su-232106DiVA, id: diva2:1885718
Conference
8th European Conference of the Prognostics and Health Managements Society, 3-5 July 2024, Prague, Czech Republic.
Available from: 2024-07-24 Created: 2024-07-24 Last updated: 2024-07-29Bibliographically approved

Open Access in DiVA

No full text in DiVA

Other links

Publisher's full text

Authority records

Kharazian, Zahra

Search in DiVA

By author/editor
Kharazian, Zahra
By organisation
Department of Computer and Systems Sciences
Computer Sciences

Search outside of DiVA

GoogleGoogle Scholar

doi
isbn
urn-nbn

Altmetric score

doi
isbn
urn-nbn
Total: 136 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf