Enhancing variational quantum state diagonalization using re