It would be helpful to have the option to bake an ID map from your materials but get random colors, like the other source options, rather than the materials base color. My high poly it setup with materials that are an approximation of what the want the final textures to look like, this means some of the materials share very similair base colors, which makes them very difficult to work with as ID map colors.