greenelab / mpmp

Multimodal Pan-cancer Mutation Prediction
BSD 3-Clause "New" or "Revised" License
7 stars 6 forks source link

Multi-omics neural network models, part 1 #84

Closed jjc2718 closed 2 years ago

jjc2718 commented 2 years ago

One of our reviewers asked us to try a neural network model for the multi-omics mutation prediction problem, to see if a more complex model can pick up on interactions between different -omics layers.

This PR is a bit of a "pilot" experiment in the sense that I'm fitting pretty small models (1000 PCs for each -omics type, fairly small hidden layers, only 4 different genes) in order to get things to train quickly. The idea was that if we struggled with getting this to outperform elastic net, it might not be worth the effort/computational resources of fitting larger models on higher input dimensionality, etc.

However, it seems like the neural network tends to give a fairly clear performance improvement over elastic net also using 1000 PCs, particularly in the multi-omics case:

image

So, we'll continue to pursue this, to make a comparison on the same data/genes as we have in the paper.

Code changes: