I think that is the solution. Send a memo that the league will be watching tape, and players that consistently flop or offensive players that cause unnatural offensive contact will be watched, and the benefit of the doubt will start going to the defender for those players.
I also think as part of a challenge, if it is determined that a player clearly flopped which led the officials to an incorrect call, it should be two shots and the ball for the defending team and a personal foul for the offender. This would make the challenge (which is neutered for many situations due to leading to a jump ball) more effective.