Improved Generalization Guarantees in Restricted Data Models