Systems and methods for treatment selection
There is disclosed a method and a system for predicting the efficacy of one or more treatments. A completed questionnaire may be received from a patient requiring treatment. The responses to the questionnaire may be input to a machine learning algorithm (MLA). The MLA may have been trained using labelled patient data. A predicted efficacy of one or more treatments and a prototype corresponding to the patient may be received from the MLA. An interface may be output indicating the predicted efficacy of the one or more treatments and the prototype.
1. A method for predicting a treatment efficacy for a patient, the method comprising:
training a machine learning algorithm (MLA) by:
receiving datasets from one or more sources corresponding to treatments for mental illness, wherein each data point in the datasets comprises questionnaire data corresponding to a patient and an indication of treatment efficacy corresponding to the respective patient;
normalizing results of the datasets, thereby generating normalized results;
generating, based on the normalized results, a training dataset;
selecting one or more features in the training dataset;
training, using the selected one or more features, the MLA to predict, for input patient data, an efficacy of each of a plurality of treatments; and
training the MLA to determine a prototype corresponding to the input patient data from a set of prototypes, wherein each prototype of the set of prototypes corresponds to a cluster of patient data, wherein the training is performed using a loss function, wherein the loss function determines a distance between prototypes, and wherein the loss function determines a variance in remission predictions between the prototypes, and
predicting the treatment efficacy for the patient by:
receiving questionnaire responses from the patient;
inputting the questionnaire responses into the MLA;
outputting, by the MLA, a predicted efficacy of each of the plurality of treatments for the patient;
outputting, by the MLA, a prototype of the set of prototypes corresponding to the patient;
generating, based on the predicted efficacy of the one or more treatments and the prototype, an interface; and
outputting for display the interface.
2. The method of claim 1 , wherein the interface comprises, for each of the one or more treatments, a predicted likelihood of remission.
3. The method of claim 1 , further comprising receiving, via the interface, user input indicating a treatment plan, wherein the treatment plan comprises at least one of the one or more treatments.
4. The method of claim 1 , wherein the loss function is configured to increase the distance between prototypes and increase the variance in remission predictions between the prototypes.
5. The method of claim 1 , wherein the prototype corresponding to the patient indicates a cluster of patients having similar features to the patient, wherein the interface provides an indication of a similarity between the patient and the cluster of patients, and wherein the interface explains how the prototype relates to the predicted efficacy.
6. The method of claim 1 , wherein the MLA comprises a neural network, wherein the prototype is determined by a layer of the neural network corresponding to prototypes, and wherein the layer of the neural network corresponding to prototypes outputs a distance between the patient and each prototype of the set of prototypes.
7. A method comprising:
receiving questionnaire responses from a patient requiring treatment;
inputting the questionnaire responses into a machine learning algorithm (MLA), wherein the MLA was trained based on labelled patient data, wherein each data point in the labelled patient data comprises questionnaire data corresponding to a respective patient and a label indicating an efficacy of a treatment for the respective patient, and wherein the MLA is configured to generate a set of prototypes based on the labelled patient data;
receiving, from the MLA, a predicted efficacy of one or more treatments for the patient;
receiving, from the MLA, a prototype of the set of prototypes corresponding to the patient;
generating, based on the predicted efficacy of the one or more treatments and the prototype, an interface; and
outputting for display the interface.
8. The method of claim 7 , wherein the interface comprises:
a predicted likelihood of remission for each of the one or more treatments, and
an indication of a distance between the patient and the prototype.
9. The method of claim 7 , wherein each prototype of the set of prototypes corresponds to a cluster of patients generated from the labelled patient data.
10. The method of claim 7 , wherein the MLA comprises a neural network, and wherein the prototype is determined by a layer of the neural network corresponding to prototypes.
11. The method of claim 10 , wherein the layer of the neural network corresponding to prototypes outputs a distance between the patient and each prototype of the set of prototypes.
12. A method for training a machine learning algorithm (MLA), the method comprising:
receiving datasets from one or more sources corresponding to treatments for mental illness, wherein each data point in the datasets comprises questionnaire data corresponding to a patient in and an indication of treatment efficacy corresponding to the respective patient;
normalizing results of the datasets, thereby generating normalized results;
generating, based on the normalized results, a training dataset;
selecting one or more features in the training dataset;
training, using the selected one or more features, the MLA to predict, for input patient data, an efficacy of each of the treatments; and
training the MLA to determine a prototype corresponding to the input patient data from a set of prototypes, wherein each prototype of the set of prototypes corresponds to a cluster of patient data, wherein the training is performed using a loss function, wherein the loss function determines a distance between prototypes of the set of prototypes, and wherein the loss function determines a variance in remission predictions between the prototypes of the set of prototypes.
13. The method of claim 12 , wherein training the MLA to determine a prototype corresponding to the input patient data comprises training the MLA based at least in part on a prototype sample distance variance.
14. The method of claim 12 , further comprising determining a prototype sample distance variance based at least in part on a variance of distances between a set of nearest samples for a given prototype and the given prototype itself.
15. The method of claim 12 , further comprising determining a prototype sample distance variance based at least in part on variance of pairwise distances between the set of prototypes.
16. The method of claim 12 , further comprising determining a prototype remission prediction based at least in part on variance of differential remission predictions for the set of prototypes across a treatment type.
17. The method of claim 12 , further comprising determining a prototype remission prediction based at least in part on variance of differential remission predictions for a given prototype across a plurality of treatment types.
18. The method of claim 12 , wherein the training is performed using a loss function, wherein the loss function determines a difference between a predicted likelihood of remission and a labeled occurrence of remission, wherein the loss function determines an autoencoder loss indicating a distance between an original sample and a decoded sample, and wherein the loss function is configured to increase the distance between prototypes and increase the variance in remission predictions between the prototypes.
19. The method of claim 12 , wherein normalizing the results of the datasets comprises grouping questions in different datasets relating to a same feature.
20. The method of claim 12 , wherein normalizing the results of the datasets comprises converting categorical responses in the datasets to binary responses.