neulab / prompt2model

prompt2model - Generate Deployable Models from Natural Language Instructions
Apache License 2.0
1.97k stars 177 forks source link

Support for non-text modalities (images, speech, video) #316

Open neubig opened 1 year ago

neubig commented 1 year ago

Currently prompt2model is limited to text input text output tasks. The underlying framework can certainly handle different modalities, and it would be great to see prompt2model be able to handle different types of tasks as well (such as image classification/generation, speech tasks, etc.).

But we'll probably need to think about several things such as:

  1. How are we picking appropriate base models and datasets for the modality
  2. What do we do about dataset generation?
  3. In the case of non-text output, how do we adjust our evaluation?

We can start discussing the necessary steps on this issue, and implement the necessary pieces bit-by-bit. We'd be happy for contributions!

MahamedDucale commented 1 year ago

For model selection we can use an llm to determine modality from user prompt and then retrieve appropriate dataset and model.

I think dataset generation would entail using another model retriever module to select a generative model for the modality of interest, that is if model performance increases only and if not dataset retrieval would only be used.

In the case of non text ouptut evaluation, perhaps using an appropriate evaluation metric using the huggingface evaluation library which is also retrieved.

zhaochenyang20 commented 1 year ago

Cool. Some HCI faculties in Tsinghua also talked me with multi-modalities Prompt2Model.

pieris98 commented 9 months ago

For other modalities (e.g. visual QA, video anomaly detection, image generation, speech-to-text, text-to-speech etc.), it would be nice for start to just propose existing datasets and/or models, since prompt2model is advertised as a better solution for retrieval of datasets/models than search engines and human manual searching.

neubig commented 9 months ago

Thanks @pieris98 ! In theory it should already be able to do this, but we might have to include these datasets in the dataset index. CC @ritugala who has recently re-created the dataset index and might be able to give additional guidance.