adding sae_b feedback earlier in the decoder