Adding geodesic information and stochastic patch-wise image prediction for small dataset learning - Archive ouverte HAL Accéder directement au contenu
Article Dans Une Revue Neurocomputing Année : 2021

Adding geodesic information and stochastic patch-wise image prediction for small dataset learning

Adam Hammoumi
  • Fonction : Auteur
  • PersonId : 1073777
Sylvain Desroziers

Résumé

Most recent methods of image augmentation and prediction are building upon the deep learning paradigm. A careful preparation of the image dataset and the choice of a suitable network architecture are crucial steps to assess the desired image features and, thence, achieve accurate predictions. We first propose to help the learning process by adding structural information with specific distance transform to the input image data. To handle cases with limited number of training samples, we propose a patch-based procedure with a stratified sampling method at inference. We validate our approaches on two image datasets, corresponding to two different tasks. The ability of our method to segment and predict images is investigated through the ISBI 2012 segmentation challenge dataset and generated electric field masks, respectively. The obtained results are evaluated using appropriate metrics: VRand for image segmentation and SSIM, UIQ and PSNR for image prediction. The proposed techniques demonstrate that the established framework is a reliable estimation method that could be used for a wide range of applications.
Fichier principal
Vignette du fichier
1-s2.0-S092523122100196X-main.pdf (3.78 Mo) Télécharger le fichier
Origine : Fichiers éditeurs autorisés sur une archive ouverte

Dates et versions

hal-02879709 , version 1 (24-06-2020)
hal-02879709 , version 2 (04-12-2020)
hal-02879709 , version 3 (06-09-2021)

Licence

Paternité - Pas d'utilisation commerciale - Pas de modification

Identifiants

Citer

Adam Hammoumi, Maxime Moreaud, Christophe Ducottet, Sylvain Desroziers. Adding geodesic information and stochastic patch-wise image prediction for small dataset learning. Neurocomputing, 2021, 456, pp.481-491. ⟨10.1016/j.neucom.2021.01.108⟩. ⟨hal-02879709v3⟩
5078 Consultations
162 Téléchargements

Altmetric

Partager

Gmail Facebook X LinkedIn More