Best practical method of knowledge distillation available?

TL;DR: Knowledge distillation generally performs worse than traning model from scratch on data from what I’ve seen online. Is there a method of KD where this doesn’t happen and I get close to performance of a model if it was trained from scratch?

So I’ve recently been interested in make DL models more useful for everyday tasks. And considering their size trying to run these models on consumer devices without much loss in quality but rn from what I’ve seen, this just feels like trying fit an elephant into his pants.

Basically it tears everytime I try. I found quantization to be cool but I need to reduce its size even more tbh. So I found knowledge distillation. But from what I’ve seen, though theoretically it is fantastic. Practically knowlege distillation sucks. And is probably worse than just straight up traning the model from scratch on the dataset.

So is there a used and proven method of knowledge distillation that I can use? Which will give me at least very close accuracy to a model trained from scratch on dataset?

  • PaganPasta@alien.topBEnglish
    1·
    1 year ago

    I’ve used KD and it has always performed better than training from scratch for me.

    Could be a data related issue(scarcity, quality)? Or maybe you need to find good hyper-params.

    To which domain are you applying it ?

    • Xanta_Kross@alien.topOPBEnglish
      1·
      1 year ago

      NLP. I’m trying to take the llama2 chat and try to compress it down so that it can be ran in a mid-high cpu without losing too much accuracy.

      • PaganPasta@alien.topBEnglish
        1·
        1 year ago

        What are your losses/objectives? I’m inclined to say you need a huge amount of data for it as well.

        • Xanta_Kross@alien.topOPBEnglish
          1·
          1 year ago

          I haven’t exactly chosen my specific loss function yet. From what people have told me looking up iBots loss and DinoV2’s loss as well as a loss from a paper by Google might be helpful I think. But I might just end up summing multiple loss functions if they’re useful and then check if they work.

          As for my objective, I don’t really have a specific application in my mind rn other than a chatbot of sorts (with moderate-high capability of logic/reasoning) but on my CPU.

          Currently this is a rough idea of how I want it to work tbh:

          1. Write a query about what it needs to find the answer for given question from the web
          2. Know when it has obtained the info from the web after looking up the first link and reading its contents otherwise discard it and change query and try again.
          3. After finding it’s content answer the asked common sense/logical reasoning question.

          E.g: Q. How should I take a rectangle door outside if all I have is a square window? Possible queries:

          1. Can rectangle fit into square?
          2. Rectangles shape
          3. Squares shape
          4. Standard window size
          5. Standard door size Etc

          Possible/acceptable answers:

          1. Sorry from what I’ve seen I couldn’t find the answer. (This option would be choosen if the model doesn’t find the answer in a limit of n queries)

          2. Rectangles are more general than squares and windows are generally smaller than doors so depending on your exact size you might just be able to fit it through but if the door size and window size are anything standard I don’t think you’ll be able to fit it through.