zebra-kb-wg-train

zebra-kb-wg-train

sapienzanlp

ZEBRA is a zero-shot retrieval framework for commonsense QA, built on E5-base-v2, offering significant accuracy improvements across multiple datasets.

PropertyValue
Base Modelintfloat/e5-base-v2
LicenseCreative Commons
LanguageEnglish
PaperarXiv:2410.05077

What is zebra-kb-wg-train?

ZEBRA-KB-WG-Train is a specialized knowledge base component of the ZEBRA framework, designed for zero-shot example-based retrieval augmentation in commonsense question answering. It's specifically trained on the Winogrande dataset and represents a crucial part of ZEBRA's three-stage pipeline: example retrieval, knowledge generation, and informed reasoning.

Implementation Details

This model builds upon the E5-base-v2 architecture and implements a sophisticated retrieval system that enables zero-shot learning capabilities. The implementation focuses on efficient knowledge retrieval and generation for commonsense reasoning tasks.

  • Built on the E5-base-v2 foundation model
  • Implements a three-stage pipeline architecture
  • Specialized for Winogrande training data
  • Supports zero-shot learning capabilities

Core Capabilities

  • Example-based retrieval for commonsense questions
  • Knowledge generation from retrieved examples
  • Zero-shot reasoning capabilities
  • Integration with various LLM backends
  • Significant performance improvements (60.7% accuracy on Winogrande dataset)

Frequently Asked Questions

Q: What makes this model unique?

ZEBRA's unique approach lies in its ability to perform zero-shot commonsense reasoning without task-specific training, utilizing a novel three-stage pipeline that combines example retrieval with knowledge generation.

Q: What are the recommended use cases?

This model is specifically designed for commonsense question answering tasks, particularly those requiring complex reasoning and understanding of everyday situations. It's especially effective when integrated with larger language models like Mistral-7B or Llama-3.

Related Models

Socials
PromptLayer
Company
All services online
Location IconPromptLayer is located in the heart of New York City
PromptLayer © 2026