Entanglement entropy from non-equilibrium Monte Carlo simulations