I know this is not exactly what you’re asking for, but just as an FYI, I’ve been getting into AI upscaling of classic 80s/90s hand-drawn content using chaiNNer and some of the models on https://openmodeldb.info/, and they have some really impressive cleanup models on there that can help with things like dot crawl, rainbowing, haloing, etc (Dotzilla in particular). The results put Topaz AI to shame.

chaiNNer also has an “Average Color Fix” node that lets you compare an image to a reference image, and it will take the colors from the reference image and apply them to the image. It’s really impressive stuff.

There are three images in the following comparison: The source (sized up), the upscale (colors have shifted), and upscale (color-corrected using Average Color Fix)
https://imgsli.com/MjE5MDg0/0/1

If you think learning about this process would help you, give me a shout and I can get you more details on how to do it. Otherwise, good luck!