I can recomand you some related papers which focus on using attention method to select important for all MRI slices.Interpretable medical deep framework by logits-constraint attention guiding graph-based multi-scale fusion for Alzheimer’s disease analysis and Cross-attention guided loss-based deep dual-branch fusion network for liver tumor classification these two papers focus on this.You can check them.