Isn't this solvable by creating training data that says B is A? Instead of always training it on A is B.