abstract dropout_below to mask_state_below