Ajouter à un algorithme une "attention gate" pour segmenter un volume

Bonjour,
Je souhaite améliorer l’algorithme de segmentation de l’oreille interne disponible ici : [IEVNet]([URL visible pour les membres Pro]), en y intégrant une méthode avancée comme une porte d’attention (attention gate) ou toute autre approche pertinente.
Les portes d’attention permettent de mieux cibler les régions d’intérêt, ici l’oreille interne, tout en supprimant les zones de fond non pertinentes. Leur fonctionnement est bien illustré dans cet exemple : [Attention-Gated Networks]([URL visible pour les membres Pro]).

À titre d’exemple, un algorithme appliquant ce type de mécanisme pour segmenter l’oreille interne est présenté dans cet article : [Nature - Automatic segmentation of the inner ear]([URL visible pour les membres Pro]).