Calculation of mass transfer in a back-flow model with non-linear equilibrium and variable back-flow