yi-bagel-2x34b

Released January 11, 2024

This is a merge of pre-trained language models created using mergekit. For more information, kindly refer to the model cards from jondurbin linked in the section below. This model debuted in the leaderboard at rank #4 (January 11, 2024).

Merge Details

Merge Method

This model is an expertimental merge using the linear merge method. This is to assess the degree of which the DPO has an effect, in terms of censoring, as used in jondurbin/bagel-dpo-34b-v0.2.

Models Merged

The following models were included in the merge:

• jondurbin/bagel-dpo-34b-v0.2
• jondurbin/nontoxic-bagel-34b-v0.2

Open LLM Leaderboard Metrics (as of January 11, 2024)

Metric	Value
MMLU (5-shot)	76.60
ARC (25-shot)	72.70
HellaSwag (10-shot)	85.44
TruthfulQA (0-shot)	71.42
Winogrande (5-shot)	82.72
GSM8K (5-shot)	60.73
Average	74.93

According to the leaderboard description, here are the benchmarks used for the evaluation:

• MMLU (5-shot) - a test to measure a text model’s multitask accuracy. The test covers 57 tasks including elementary mathematics, US history, computer science, law, and more.
• AI2 Reasoning Challenge -ARC- (25-shot) - a set of grade-school science questions.
• HellaSwag (10-shot) - a test of commonsense inference, which is easy for humans (~95%) but challenging for SOTA models.
• TruthfulQA (0-shot) - a test to measure a model’s propensity to reproduce falsehoods commonly found online.
• Winogrande (5-shot) - an adversarial and difficult Winograd benchmark at scale, for commonsense reasoning.
• GSM8k (5-shot) - diverse grade school math word problems to measure a model's ability to solve multi-step mathematical reasoning problems.

Configuration

The following YAML configuration was used to produce this model:

models:
  - model: jondurbin/nontoxic-bagel-34b-v0.2
    parameters:
      weight: 0.5
  - model: jondurbin/bagel-dpo-34b-v0.2
    parameters:
      weight: 0.5
merge_method: linear
dtype: float16

Further Information

For additional information or inquiries about yi-bagel-2x34b, please contact the developer through email: jasperkylecatapang@gmail.com.