Toxicity prediction in cancer using multiple instance learning in a multi-task framework

Cheng Li, Sunil Gupta, Santu Rana, Wei Luo, Svetha Venkatesh, David Ashely, Dinh Phung

Research output: Chapter in Book/Report/Conference proceedingConference PaperResearchpeer-review

2 Citations (Scopus)

Abstract

Treatments of cancer cause severe side effects called toxicities. Reduction of such effects is crucial in cancer care. To impact care, we need to predict toxicities at fortnightly intervals. This toxicity data differs from traditional time series data as toxicities can be caused by one treatment on a given day alone, and thus it is necessary to consider the effect of the singular data vector causing toxicity. We model the data before prediction points using the multiple instance learning, where each bag is composed of multiple instances associated with daily treatments and patient-specific attributes, such as chemotherapy, radiotherapy, age and cancer types. We then formulate a Bayesian multi-task framework to enhance toxicity prediction at each prediction point. The use of the prior allows factors to be shared across task predictors. Our proposed method simultaneously captures the heterogeneity of daily treatments and performs toxicity prediction at different prediction points. Our method was evaluated on a real-word dataset of more than 2000 cancer patients and had achieved a better prediction accuracy in terms of AUC than the state-of-art baselines.

Original languageEnglish
Title of host publicationAdvances in Knowledge Discovery and Data Mining
Subtitle of host publication20th Pacific-Asia Conference, PAKDD 2016 Auckland, New Zealand, April 19–22, 2016 Proceedings, Part I
EditorsJames Bailey, Latifur Khan, Takashi Washio, Gillian Dobbie, Joshua Zhexue Huang, Ruili Wang
Place of PublicationCham Switzerland
PublisherSpringer
Pages152-164
Number of pages13
ISBN (Electronic)9783319317533
ISBN (Print)9783319317526
DOIs
Publication statusPublished - 2016
Externally publishedYes
EventPacific-Asia Conference on Knowledge Discovery and Data Mining 2016 - Auckland, New Zealand
Duration: 19 Apr 201622 Apr 2016
Conference number: 20th
http://pakdd16.wordpress.fos.auckland.ac.nz/
https://link.springer.com/book/10.1007/978-3-319-31753-3 (Proceedings)

Publication series

NameLecture Notes in Artificial Intelligence
PublisherSpringer
Volume9651
ISSN (Print)0302-9743
ISSN (Electronic)1611-3349

Conference

ConferencePacific-Asia Conference on Knowledge Discovery and Data Mining 2016
Abbreviated titlePAKDD 2016
Country/TerritoryNew Zealand
CityAuckland
Period19/04/1622/04/16
Internet address

Cite this