fix fatal bug: zero-out iim_diff in backward