Etsc-Net: A Parameter-efficient Convolutional Neural Network for Drug Treatment Outcome Studies of Pediatric Epilepsy

Abstract Background: Ability to predict the outcomes of pharmacological treatment of epilepsy in pediatric patients with tuberous sclerosis complex (TSC) can confer a distinct leverage and guide therapeutic decision-making. Multi-contrast magnetic resonance imaging (MRI) is routinely used for diagnosis of TSC by pediatricians. We propose a parameter-efficient convolutional neural network with multi-contrast images to predict the drug treatment outcomes of pediatric epilepsy in TSC. Methods: Image-based models were generated using the EfficientNet3D-B0 network architecture. A weighted average ensemble network with multi-contrast images was created as the final model. The proposed neural network is named as Efficient Tuberous sclerosis complex-Net (eTSC-Net).We compared our methods with a Residual Network 3D（ResNet3D) model. We trained a 3D-ResNet on our T2FLAIR data. Binary classification models were trained to distinguish non-controlled group patients from controlled group patients on T2W and T2FLAIR images. We trained all the models using an Nvidia RTX A6000 Graphical Processing Unit (GPU) card. Area under curve(AUC), accuracy (ACC), sensitivity (SEN), and specificity (SPE) were calculated to assess the classification performance for each model in each cohort. The differences between subgroups were assessed using independent samples t test and pvalues < 0.05 were considered indicative of statistical significance. Results: The proposed neural network (eTSC-Net) achieved the best performance with an AUC value of 0.833 and 90.0% accuracy in the testing cohort, which was better than other models. Conclusions: The results demonstrated the ability of the proposed method for predicting drug treatment outcomes in pediatric TSC-related epilepsy. eTSC-Net can serve as a useful computer-aided diagnostic tool to help clinical radiologists formulate more targeted treatment.