BonsaiBONSAI
事例Case study

Yandex

Research on language model sensitivity to writing style

Duration4 months
Yandex
+8 pp
Precision@10
概要About the project01

The research let us fine-tune the model into an analysis engine that can tell whether texts were generated by the same language model or written by the same person.

課題Key challenge02

Research uncertainty: hypotheses were tested iteratively with no guaranteed outcome. A limited compute budget for fine-tuning.

要点The solution

We carried out applied research on multilingual-e5-base: we studied the language model’s sensitivity to writing styles and proposed a fine-tuning method that reduces errors in cross-lingual tasks. The research produced a Gabriel Graph that shows which authors wrote similar texts. If the model judged two texts “mutually sensitive,” the graph linked their authors. The results have been integrated into several of the client’s language models and serve as a “temperature” indicator of how interconnected the data is.

工程Process03
  1. 0101

    Problem research

    Reviewing existing approaches to text style extraction and metric-learning architectures.

  2. 0202

    Dataset preparation

    Assembling a text corpus and labeling authorship and style features.

  3. 0303

    Neural network architecture

    Building models with ArcFace, SphereFace, and CosFace loss functions.

  4. 0404

    Contrastive learning

    Training the system to extract stylistic features of text with contrastive approaches.

  5. 0505

    Quality evaluation

    Measuring author identification accuracy on short text fragments.

  6. 0606

    Research write-up

    Compiling the results and extending the work toward detecting AI-generated text.

成果Results04
+8 ppPrecision@10
−23%cross-lingual errors
6languages studied
記録Project gallery05
技術Tech stack
  • 01PyTorch
  • 02HuggingFace
  • 03multilingual-e5-base
  • 04Python
  • 05Streamlit
  • 06CUDA
  • 07TensorBoard
  • 08LangChain

Have a similar challenge?

We’ll review it and estimate within 5 business days.

Discuss your project