{"status":"ok","message-type":"work","message-version":"1.0.0","message":{"indexed":{"date-parts":[[2026,3,3]],"date-time":"2026-03-03T16:11:54Z","timestamp":1772554314493,"version":"3.50.1"},"reference-count":35,"publisher":"MDPI AG","issue":"2","license":[{"start":{"date-parts":[[2021,2,17]],"date-time":"2021-02-17T00:00:00Z","timestamp":1613520000000},"content-version":"vor","delay-in-days":0,"URL":"https:\/\/creativecommons.org\/licenses\/by\/4.0\/"}],"content-domain":{"domain":[],"crossmark-restriction":false},"short-container-title":["J. Imaging"],"abstract":"<jats:p>We present a sample-efficient image segmentation method using active learning, we call it Active Bayesian UNet, or AB-UNet. This is a convolutional neural network using batch normalization and max-pool dropout. The Bayesian setup is achieved by exploiting the probabilistic extension of the dropout mechanism, leading to the possibility to use the uncertainty inherently present in the system. We set up our experiments on various medical image datasets and highlight that with a smaller annotation effort our AB-UNet leads to stable training and better generalization. Added to this, we can efficiently choose from an unlabelled dataset.<\/jats:p>","DOI":"10.3390\/jimaging7020037","type":"journal-article","created":{"date-parts":[[2021,2,17]],"date-time":"2021-02-17T21:35:42Z","timestamp":1613597742000},"page":"37","update-policy":"https:\/\/doi.org\/10.3390\/mdpi_crossmark_policy","source":"Crossref","is-referenced-by-count":23,"title":["Active Learning with Bayesian UNet for Efficient Semantic Image Segmentation"],"prefix":"10.3390","volume":"7","author":[{"ORCID":"https:\/\/orcid.org\/0000-0002-3064-2093","authenticated-orcid":false,"given":"Isah Charles","family":"Saidu","sequence":"first","affiliation":[{"name":"Department of Computer Science, African University of Science and Technology, Abuja, Nigeria"}]},{"given":"Lehel","family":"Csat\u00f3","sequence":"additional","affiliation":[{"name":"Depterment of Mathematics and Informatics, Babe\u015f\u2013Bolyai University, RO-400084 Cluj-Napoca, Romania"}]}],"member":"1968","published-online":{"date-parts":[[2021,2,17]]},"reference":[{"key":"ref_1","doi-asserted-by":"crossref","unstructured":"Szeliski, R. (2010). Computer Vision: Algorithms and Applications, Springer.","DOI":"10.1007\/978-1-84882-935-0"},{"key":"ref_2","doi-asserted-by":"crossref","unstructured":"Hu, Z., Zou, Q., and Li, Q. (2015, January 27\u201330). Watershed superpixel. Proceedings of the 2015 IEEE International Conference on Image Processing (ICIP), Quebec City, QC, Canada.","DOI":"10.1109\/ICIP.2015.7350818"},{"key":"ref_3","doi-asserted-by":"crossref","first-page":"888","DOI":"10.1109\/34.868688","article-title":"Normalized cuts and image segmentation","volume":"22","author":"Shi","year":"2000","journal-title":"IEEE Trans. Pattern Anal. Mach. Intell."},{"key":"ref_4","unstructured":"Koller, D., Friedman, N., Getoor, L., and Taskar, B. (2019, October 03). Graphical Models in a Nutshell. Available online: http:\/\/citeseerx.ist.psu.edu\/viewdoc\/summary?doi=10.1.1.146.2935."},{"key":"ref_5","doi-asserted-by":"crossref","unstructured":"Vezhnevets, A., Ferrari, V., and Buhmann, J. (2011, January 6\u201313). Weakly Supervised Semantic Segmentation with Multi Image Model. Proceedings of the 2011 International Conference on Computer Vision, Barcelona, Spain.","DOI":"10.1109\/ICCV.2011.6126299"},{"key":"ref_6","doi-asserted-by":"crossref","first-page":"470","DOI":"10.1109\/TMM.2013.2293424","article-title":"Representative discovery of structure cues for weakly-supervised image segmentation","volume":"16","author":"Zhang","year":"2014","journal-title":"IEEE Trans. Multimed."},{"key":"ref_7","doi-asserted-by":"crossref","unstructured":"Konyushkova, K., Sznitman, R., and Fua, P. (2015, January 7\u201313). Introducing geometry in active learning for image segmentation. Proceedings of the IEEE International Conference on Computer Vision, Santiago, Chile.","DOI":"10.1109\/ICCV.2015.340"},{"key":"ref_8","first-page":"282","article-title":"Conditional random fields: Probabilistic models for segmenting and labeling sequence data","volume":"8","author":"Lafferty","year":"2001","journal-title":"Proc. ICML"},{"key":"ref_9","unstructured":"Krizhevsky, A., Sutskever, I., and Hinton, G.E. (2012). ImageNet Classification with Deep Convolutional Neural Networks. Proceedings of the NIPS\u201912: 25th International Conference on Neural Information Processing Systems-Volume 1, Curran Associates Inc."},{"key":"ref_10","unstructured":"Pereira, F., Burges, C.J.C., Bottou, L., and Weinberger, K.Q. (2012). Deep Neural Networks Segment Neuronal Membranes in Electron Microscopy Images. Advances in Neural Information Processing Systems 25, Curran Associates, Inc."},{"key":"ref_11","unstructured":"Cortes, C., Lawrence, N.D., Lee, D.D., Sugiyama, M., and Garnett, R. (2015). Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks. Advances in Neural Information Processing Systems 28, Curran Associates, Inc."},{"key":"ref_12","doi-asserted-by":"crossref","unstructured":"He, K., Gkioxari, G., Doll\u00e1r, P., and Girshick, R.B. (2017, January 22\u201329). Mask R-CNN. Proceedings of the IEEE International Conference on Computer Vision, Venice, Italy.","DOI":"10.1109\/ICCV.2017.322"},{"key":"ref_13","doi-asserted-by":"crossref","unstructured":"Navab, N., Hornegger, J., Wells, W., and Frangi, A. (2015). U-Net: Convolutional Networks for Biomedical Image Segmentation. Medical Image Computing and Computer-Assisted Intervention\u2014MICCAI 2015, Springer. Lecture Notes in Computer Science.","DOI":"10.1007\/978-3-319-24553-9"},{"key":"ref_14","doi-asserted-by":"crossref","first-page":"640","DOI":"10.1109\/TPAMI.2016.2572683","article-title":"Fully Convolutional Networks for Semantic Segmentation","volume":"39","author":"Shelhamer","year":"2014","journal-title":"IEEE Trans. Pattern Anal. Mach. Intell."},{"key":"ref_15","unstructured":"Arbib, M.A. (1995). Convolutional Networks for Images, Speech and Time Series. The Handbook of Brain Theory and Neural Networks, The MIT Press."},{"key":"ref_16","unstructured":"Gal, Y. (2016). Uncertainty in Deep Learning. [Ph.D. Thesis, University of Cambridge]."},{"key":"ref_17","unstructured":"Gal, Y., and Ghahramani, Z. (2016, January 20\u201322). Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learning. Proceedings of the 33rd ICML, New York, NY, USA."},{"key":"ref_18","unstructured":"Teye, M., Azizpour, H., and Smith, K. (2018, January 10\u201315). Bayesian Uncertainty Estimation for Batch Normalized Deep Networks. Proceedings of the 35th ICML, Stockholm, Sweden."},{"key":"ref_19","first-page":"448","article-title":"Batch Normalization: Accelerating Deep Network Training by Reducing Internal Covariate Shift","volume":"Volume 37 (ICML\u201915)","author":"Ioffe","year":"2015","journal-title":"International Conference on Machine Learning"},{"key":"ref_20","doi-asserted-by":"crossref","unstructured":"Vezhnevets, A., Buhmann, J.M., and Ferrari, V. (2012, January 16\u201321). Active learning for semantic segmentation with expected change. Proceedings of the 2012 IEEE Conference on Computer Vision and Pattern Recognition, Providence, RI, USA.","DOI":"10.1109\/CVPR.2012.6248050"},{"key":"ref_21","unstructured":"Fathi, A., Balcan, M.F., Ren, X., and Rehg, J.M. (September, January 29). Combining Self Training and Active Learning for Video Segmentation. Proceedings of the British Machine Vision Conference, Dundee, UK."},{"key":"ref_22","first-page":"5574","article-title":"What Uncertainties Do We Need in Bayesian Deep Learning for Computer Vision","volume":"Volume 30","author":"Guyon","year":"2017","journal-title":"Advances in Neural Information Processing Systems"},{"key":"ref_23","doi-asserted-by":"crossref","unstructured":"Mahapatra, D., Bozorgtabar, B., Thiran, J., and Reyes, M. (2018). Efficient Active Learning for Image Classification and Segmentation using a Sample Selection and Conditional Generative Adversarial Network. International Conference on Medical Image Computing and Computer-Assisted Intervention, Springer.","DOI":"10.1007\/978-3-030-00934-2_65"},{"key":"ref_24","unstructured":"Gorriz, M., Carlier, A., Faure, E., and Gir\u00f3 i Nieto, X. (2021, February 01). Cost-Effective Active Learning for Melanoma Segmentation. Available online: https:\/\/ui.adsabs.harvard.edu\/abs\/2017arXiv171109168G."},{"key":"ref_25","doi-asserted-by":"crossref","unstructured":"Saidu, C.I., and Csat\u00f3, L. (2019). Medical Image Analysis with Semantic Segmentation and Active Learning. Stud. Univ. Babes-Bolyai Inform., 64.","DOI":"10.24193\/subbi.2019.1.03"},{"key":"ref_26","doi-asserted-by":"crossref","first-page":"321","DOI":"10.1613\/jair.953","article-title":"SMOTE: Synthetic Minority Over-sampling Technique","volume":"16","author":"Chawla","year":"2002","journal-title":"J. Artif. Intell. Res."},{"key":"ref_27","unstructured":"Houlsby, N., Husz\u00e1r, F., Ghahramani, Z., and Lengyel, M. (2011). Bayesian Active Learning for Classification and Preference Learning. arXiv."},{"key":"ref_28","unstructured":"Kingma, D.P., and Ba, J. (2021, February 01). Adam: A Method for Stochastic Optimization. Available online: http:\/\/arxiv.org\/abs\/1412.6980."},{"key":"ref_29","first-page":"1929","article-title":"Dropout: A Simple Way to Prevent Neural Networks from Overfitting","volume":"15","author":"Srivastava","year":"2014","journal-title":"J. Mach. Learn. Res."},{"key":"ref_30","doi-asserted-by":"crossref","first-page":"297","DOI":"10.2307\/1932409","article-title":"Measures of the Amount of Ecologic Association Between Species","volume":"26","author":"Dice","year":"1945","journal-title":"Ecology"},{"key":"ref_31","first-page":"1","article-title":"A Method of Establishing Groups of Equal Amplitude in Plant Sociology Based on Similarity of Species Content and Its Application to Analyses of the Vegetation on Danish Commons","volume":"5","year":"1948","journal-title":"Biol. Skar."},{"key":"ref_32","first-page":"142","article-title":"Crowdsourcing the creation of image segmentation algorithms for connectomics","volume":"9","author":"Turaga","year":"2015","journal-title":"Front. Neuroanat."},{"key":"ref_33","doi-asserted-by":"crossref","first-page":"489","DOI":"10.1016\/j.media.2016.08.008","article-title":"Gland Segmentation in Colon Histology Images: The GlaS Challenge Contest","volume":"35","author":"Sirinukunwattana","year":"2016","journal-title":"Medical Image Analysis."},{"key":"ref_34","first-page":"201","article-title":"Active Learning Literature Survey","volume":"15","author":"Settles","year":"2010","journal-title":"Mach. Learn."},{"key":"ref_35","unstructured":"MacKay, D.J.C. (2002). Information Theory, Inference & Learning Algorithms, Cambridge University Press."}],"container-title":["Journal of Imaging"],"original-title":[],"language":"en","link":[{"URL":"https:\/\/www.mdpi.com\/2313-433X\/7\/2\/37\/pdf","content-type":"unspecified","content-version":"vor","intended-application":"similarity-checking"}],"deposited":{"date-parts":[[2025,10,11]],"date-time":"2025-10-11T05:25:12Z","timestamp":1760160312000},"score":1,"resource":{"primary":{"URL":"https:\/\/www.mdpi.com\/2313-433X\/7\/2\/37"}},"subtitle":[],"short-title":[],"issued":{"date-parts":[[2021,2,17]]},"references-count":35,"journal-issue":{"issue":"2","published-online":{"date-parts":[[2021,2]]}},"alternative-id":["jimaging7020037"],"URL":"https:\/\/doi.org\/10.3390\/jimaging7020037","relation":{},"ISSN":["2313-433X"],"issn-type":[{"value":"2313-433X","type":"electronic"}],"subject":[],"published":{"date-parts":[[2021,2,17]]}}}