I really think it is rather simple.
Change the wording to reflect the NBA interp.
If the defense knocks the ball loose, no team control.
The only other problem is recognizing the foul when it happens as an offensive foul, not just a "loose ball type foul." This change also makes it so that a ball knocked loose by the defense which is then touched last by an offensive player, can be recovered by the offense in the backcourt with no violation. Sweet!