“…Despite general improvements of forecasts, they tend to smooth the extreme precipitation at sub-seasonal scales (Baño-Medina et al, 2021;Kim et al, 2022), likely due to insufficient heavy precipitation samples (Chen et al, 2022). Many studies have since introduced more recent variants of CNNs including U-Net (Ni et al, 2023) and SmaAt-UNet (Li et al, 2024), or coupled standard CNNs with different structures, such as Auto-Encoder (Ling et al., 2022), Transformer (Ling et al, 2024), and in particular ResNet, which shows the potential of mitigating the vanishing gradient issue by introducing the residual paths (Nie et al, 2024). Others have attempted to introduce specialized loss functions to balance heavy and light rains, such as the exponentially weighted mean squared error (Ebert-Uphoff et al, 2020) and Dice loss (You et al, 2022).…”