Multiple sclerosis detection via 6-layer stochastic pooling convolutional neural network and multiple-way data augmentation

  • Jian Wang School of Computing and Mathematical Sciences, University of Leicester, Leicester, LE1 7RH, UK
  • Dimas Lima 2 Department of Electrical Engineering, Federal University of Santa Catarina, 88040-900, Florianópolis, Brazil
Keywords: Multiple sclerosis, Convolutional neural network, Stochastic pooling, Data augmentation, Magnetic resonance imaging


Background: Multiple sclerosis is one of most widespread autoimmune neuroinflammatory diseases
which mainly damages body function such as movement, sensation, and vision. Despite of conventional
clinical presentation, brain magnetic resonance imaging of white matter lesions is often applied to
diagnose multiple sclerosis at the early stage.
Methods: In this article, we proposed a 6-layer stochastic pooling convolutional neural network (CNN)
with multiple-way data augmentation for multiple sclerosis detection in brain magnetic resonance imaging.
Our approach does not demand hand-crafted features unlike those traditional machine learning methods.
Via application of stochastic pooling and multiple-way data augmentation, our 6-layer CNN achieved
equivalent performance against those deep learning methods which consist of so many layers and
parameters that ordinarily bring difficulty to training. Further, we also conducted ablation experiments to
examine the contribution of stochastic pooling and multiple-way data augmentation to the original CNN
Results: The results showed that this 6-layer CNN obtained a sensitivity of 95.98 ± 0.46%, a specificity of
95.67 ± 0.92%, and an accuracy of 95.82 ± 0.58%. According to comparison experiments, our results are
better than state-of-the-art approaches.
Conclusion: Our scheme of stochastic pooling and multiple-way data augmentation enhanced the original
6-layer CNN model compared to those using maximum pooling or average pooling and inadequate data


Download data is not yet available.


. Dua M, Makhija D, Manasa PYL, Mishra P. A CNN–RNN–LSTM based amalgamation for Alzheimer’s Disease detection. J Med Biol Eng. 2020;40(5):688-706.

Si X, Zhang X, Zhou Y, Sun Y, Jin W, Yin S, et al. Automated detection of juvenile myoclonic epilepsy using CNN based transfer learning in diffusion MRI. IEEE; 2020. p.1679-82.

Lee SM, Hyeon JW, Kim S-J, Kim H, Noh R, Kim S, et al. Sensitivity and specificity evaluation of multiple neurodegenerative proteins for Creutzfeldt-Jakob disease diagnosis using a deep-learning approach. Prion. 2019;13(1):141-50.

Yang Y, Yan LF, Zhang X, Han Y, Nan HY, Hu YC, et al. Glioma grading on conventional MR images: a deep learning study with transfer learning. Front Neurosci. 2018;12:804.

Wang S., Zhan T, Chen Y, Zhang Y, Yang M, Lu H, et al. Multiple sclerosis detection based on biorthogonal wavelet transform, RBF kernel principal component analysis, and logistic regression. IEEE Access, 2016. 4: p.7567-7576.

Nayak DR, Dash R, Majhi B. Brain MR image classification using two-dimensional discrete wavelet transform and AdaBoost with random forests. Neurocomputing (Amsterdam). 2016;177:188-97.

Zhang Y-D, Pan C, Sun J, Tang C. Multiple sclerosis identification by convolutional neural network with dropout and parametric ReLU. J Comput Sci. 2018;28:1-10.

Eitel F, Soehler E, Bellmann-Strobl J, Brandt AU, Ruprecht K, Giess RM, et al. Uncovering convolutional neural network decisions for diagnosing multiple sclerosis on conventional MRI using layer-wise relevance propagation. Neuroimage Clin. 2019;24:102003.

Alijamaat A, NikravanShalmani A, Bayat P. Multiple sclerosis identification in brain MRI images using wavelet convolutional neural networks. Int J Imaging Syst Technol. 2021;31(2):778-85.

Han J, Hou S-M. Multiple sclerosis detection via wavelet entropy and feedforward neural network trained by adaptive genetic algorithm. Lect Notes Comput Sci. 2019;11507:87-97.

Han J, Hou S-M. A multiple sclerosis recognition via hu moment invariant and artificial neural network trained by particle swarm optimization. Lect Notes Inst Comput Sci Soc-Inform Telecommun Eng. 2020;327:254-64.

Tang, C. Multiple sclerosis detection via 5-layer convolutional neural network. J Med Imaging Health Inform. 2021. 11: p. 1-8.

MRI lesion segmentation in multiple sclerosis database.

Alparslan E, Fuatince M. Image enhancement by local histogram stretching. IEEE Trans Syst Man Cybern B Cybern. 1981;11:376-85.

Pasban S, Mohamadzadeh S, Zeraatkar-Moghaddam J, Shafiei AK. Infant brain segmentation based on a combination of VGG-16 and U-Net deep neural networks. IET Image Process. 2020;14(17):4756-65.

Arulmozhi P, Abirami S. DSHPoolF: deep supervised hashing based on selective pool feature map for image retrieval. Vis Comput. 2021;37(8):2391-405.

Krestinskaya O, James AP. Analogue neuro-memristive convolutional dropout nets. Proc Math Phys Eng Sci. 2020;476(2242).

Zhang YD, Zhang Z, Zhang X, Wang SH. MIDCAN: A multiple input deep convolutional attention network for Covid-19 diagnosis based on chest CT and chest X-ray. Pattern Recognit Lett. 2021;150:8-16.

Clement JC, Indira N, Vijayakumar P, Nandakumar R. Deep learning based modulation classification for 5G and beyond wireless systems. Peer Peer Netw Appl. 2021;14(1):319-32.

Wang SH, Satapathy SC, Anderson D, Chen SX, Zhang YD. Deep fractional max pooling neural network for COVID-19 recognition. Front Public Health. 2021;9:726144.

Mahmoudi MA, Chetouani A, Boufera F, Tabia H. Learnable pooling weights for facial expression recognition. Pattern Recognit Lett. 2020;138:644-50.

Wu K. SOSPCNN: structurally optimized stochastic pooling convolutional neural network for tetralogy of fallot recognition. Wirel Commun Mob Comput. 2021;2021:5792975.

Sahani M, Swain BK, Dash PK. FPGa based favourite skin colour restoration using improved histogram equalization with variable enhancement degree and ensemble extreme learning machine. IET Image Process. 2021;15:1247-59.

Lyu C, Hu G, Wang D. Attention to fine-grained information: hierarchical multi-scale network for retinal vessel segmentation. Vis Comput. 2020.

Satapathy SC, Wu D. Improving ductal carcinoma in situ classification by convolutional neural network with exponential linear unit and rank-based weighted pooling. Complex Intell Syst. 2021;7:1295-310.

Gugushvili S, Mariucci E, der Meulen F. Decompounding discrete distributions: a nonparametric Bayesian approach. Scand J Statist. 2020; 47:464-492.

Mukuno J, Matsui. Natural gradient descent of complex-valued neural networks invariant under rotations. IEICE T Fund Electr, 2019. E102A(12): p.1988-1996.

Zhang Y-D, Dong Z, Wang S-H, Yu X, Yao X, Zhou Q, et al. Advances in multimodal data fusion in neuroimaging: Overview, challenges, and novel orientation. Inf Fusion. 2020;64:149-87.

Wang SH, Zhou Q, Yang M, Zhang YD. ADVIAN: Alzheimer's disease VGG-inspired attention network based on convolutional block attention module and multiple way data augmentation. Front Aging Neurosci. 2021;13:687456.

Ryan P, Dilley B, Davies D, Glass T, Abadi F. Short-term movement patterns, population estimates and breeding biology of an island endemic bird, the Tristan Thrush. Revista Brasileira De Ornitologia, 2019. 27(4): p.245-252.

Chakraborty S, Khare K. Consistent estimation of the spectrum of trace class Data Augmentation algorithms. Bernoulli. 2019;25(4B):3832-63.

Zhang Y, Zhang X, Zhu W. ANC: attention network for COVID-19 explainable diagnosis based on convolutional block attention module. Comput Model Eng Sci. 2021;127(3):1037-58.

Shahid AI, Almotairi S. Automated Arabic sign language recognition system based on deep transfer learning. Int J Comput Sci Net Secur. 2019;19(10):144-52.

Wang SH, Zhang Y, Cheng X, Zhang X, Zhang YD. PSSPNN: PatchShuffle stochastic pooling neural network for an explainable diagnosis of COVID-19 with multiple-way data augmentation. Comput Math Methods Med. 2021;2021:6633755.

Aldabbas H, Asad M, Ryalat MH, Malik KR, Qureshi MZA. Data augmentation to stabilize image caption generation models in deep learning. Int J Adv Comput Sci Appl. 2019;10(10):571-9.

Nguyen KD, Nguyen K, Le DD, Duong DA, Nguyen TV. YADA: you always dream again for better object detection. Multimed Tools Appl. 2019;78(19):28189-208.

Zhang Y-D, Satapathy SC, Guttery DS, Górriz JM, Wang S-H. Improved breast cancer classification through combining graph convolutional network and convolutional neural network. Inf Process Manag. 2021;58(2):102439.

Pho K, Mohammed Amin MK, Yoshitaka A. Segmentation-driven hierarchical retinaNet for detecting protozoa in micrograph. Int J Semant Comput. 2019;13(03):393-413.

Wang SH, Nayak DR, Guttery DS, Zhang X, Zhang YD. COVID-19 classification by CCSHNet with deep fusion using transfer learning and discriminant correlation analysis. Inf Fusion. 2021;68:131-148.

Arvanitakis K, Avlonitis M, Papadimitriou E. Introducing stochastic recurrence interval to classification algorithms for identifying asperity patterns. Phys A: Stat Mech Appl. 2018;512:566-77.

Baquirin RBM, Fernandez PL. Artificial neural network (ANN) in a small dataset to determine neutrality in the pronunciation of English as a foreign language in Filipino Call Center Agents. Inteligencia Artificial-Iberoamerica Journal of Artificial Intelligence. 2018;21(62):134-44.

How to Cite
Wang, J., & Lima, D. (2021). Multiple sclerosis detection via 6-layer stochastic pooling convolutional neural network and multiple-way data augmentation. STEMedicine, 2(8), e101.
Research articles