Fig. 2

An overview of our proposed UniCoN. Conditional modules added to U-Net-shaped segmentation models: Conditional self-attention (ConSA) is placed at the bottleneck, and hierarchical dense spatial coordinates (HDSCs) are added to the decoder blocks. The flexibility of this design allows our approach to be encoder architecture-friendly.