Yellows or reds dished out to be rescinded also if the dive is retrospectively punished.

Minefield here though. What if the player dives, defender gets sent off, penalty ensues, and the 'innocent' team play the rest of the game with ten men and a goal behind. There will still be many willing to dive and risk a two match ban.