Chuyển đến nội dung chính

第 10 課:多模式 RAG — 文件中的圖像、表格、圖表

RAG 用於包含圖像、表格和圖表的文件。從 PDF 掃描、OCR、表格提取中提取資訊。 Vision LLM + 多模式向量搜尋。

🧠 人工智慧與機器學習 — 第 9 課 第 10 課:多模式 RAG — 圖片、表格、圖表 文檔中的地圖

真實戰鬥 RAG:從基礎到高級

第 4 部分:進階 RAG 模式

亞洲開發網

簡介

實際文件不僅僅是文字——還有圖像、表格、圖表、圖表。傳統的 RAG 會跳過這一切!多模態 RAG 解決了這個問題。

範例: 50 頁財務報告:40% 文字、30% 資料表、20% 圖表、10% 圖片。僅 RAG 文字會遺失 60% 的資訊!

本文涵蓋:

  1. 表格提取——提取表並索引
  2. 圖像理解——使用Vision LLM來描述圖像/圖形
  3. 多模態嵌入 — 將文字+圖像嵌入到同一向量空間中

1. 問題:多模式文檔

1.1 文件內容類型

┌─────────────────────────────────────────┐
│  Typical Business Document              │
│                                          │
│  [Text paragraph]                        │  ← RAG text OK
│  [Text paragraph]                        │  ← RAG text OK
│                                          │
│  ┌────────────────────────────┐          │
│  │  Revenue  │ Q1  │ Q2  │ Q3│          │  ← RAG text BỎ QUA!
│  │  Product A│ 100 │ 120 │ 95│          │
│  │  Product B│ 200 │ 180 │ 220│         │
│  └────────────────────────────┘          │
│                                          │
│  [Bar chart: Revenue trends]  📊        │  ← RAG text BỎ QUA!
│                                          │
│  [Architecture diagram]       🖼️        │  ← RAG text BỎ QUA!
│                                          │
└─────────────────────────────────────────┘

1.2 處理策略

內容類型戰略工具
文字現場直播LangChain分割器
表格擷取 → 轉換為文字/markdown非結構化,卡米洛特
圖表/圖表願景法學碩士→文字描述GPT-4o,克勞德
掃描的 PDFOCR → 文字超正方體、Azure OCR

2.表提取

2.1 使用非結構化

"""Extract tables từ PDF bằng Unstructured"""
from unstructured.partition.pdf import partition_pdf

elements = partition_pdf(
    filename="financial-report.pdf",
    strategy="hi_res",           # Dùng model detection
    infer_table_structure=True,  # Detect và extract tables
    extract_images_in_pdf=True,  # Extract images
)

# Phân loại elements
tables = []
texts = []
images = []

for el in elements:
    if el.category == "Table":
        tables.append(el)
        print(f"Table found: {el.metadata.text_as_html[:200]}...")
    elif el.category == "Image":
        images.append(el)
    else:
        texts.append(el)

print(f"Found: {len(texts)} texts, {len(tables)} tables, {len(images)} images")

2.2 表→文本摘要

"""Dùng LLM summarize bảng thành text cho RAG"""
from langchain_openai import ChatOpenAI

llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)

def summarize_table(table_html: str) -> str:
    prompt = f"""Đây là bảng dữ liệu (HTML):
{table_html}

Tóm tắt nội dung bảng thành đoạn văn (50-100 từ).
Bao gồm: tên bảng, các cột, xu hướng nổi bật, giá trị đặc biệt."""
    
    return llm.invoke(prompt).content

# Tạo document cho mỗi table
from langchain.schema import Document

table_docs = []
for table in tables:
    summary = summarize_table(table.metadata.text_as_html)
    table_docs.append(Document(
        page_content=summary,
        metadata={
            "source": "financial-report.pdf",
            "type": "table",
            "original_html": table.metadata.text_as_html,
            "page": table.metadata.page_number,
        }
    ))

2.3 多向量:同時保存摘要資料和原始數據

"""Multi-vector store: search bằng summary, trả về raw table"""
from langchain.storage import InMemoryByteStore
from langchain.retrievers.multi_vector import MultiVectorRetriever
from langchain_community.vectorstores import Chroma
from langchain_openai import OpenAIEmbeddings
import uuid

# Vector store: chứa summaries (để search)
vectorstore = Chroma(
    collection_name="multimodal",
    embedding_function=OpenAIEmbeddings(),
)

# Doc store: chứa raw data (để trả về cho LLM)
docstore = InMemoryByteStore()

retriever = MultiVectorRetriever(
    vectorstore=vectorstore,
    byte_store=docstore,
    id_key="doc_id",
)

# Index: summary → vector store, raw → doc store
for table in tables:
    doc_id = str(uuid.uuid4())
    summary = summarize_table(table.metadata.text_as_html)
    
    # Summary vào vector store (search)
    retriever.vectorstore.add_documents([
        Document(page_content=summary, metadata={"doc_id": doc_id, "type": "table"})
    ])
    
    # Raw table vào doc store (return)
    retriever.docstore.mset([(doc_id, table.metadata.text_as_html)])

💡 練習 1: 從至少包含 3 個表格的 PDF 中提取表格。建立多向量儲存:使用摘要搜索,返回原始表。測試 5 道與表格資料​​相關的問題。


3. 影像理解

3.1 Vision LLM影像描述

"""Dùng GPT-4o mô tả ảnh/biểu đồ trong tài liệu"""
import base64
from langchain_openai import ChatOpenAI

llm = ChatOpenAI(model="gpt-4o", temperature=0)

def describe_image(image_path: str) -> str:
    with open(image_path, "rb") as f:
        image_data = base64.b64encode(f.read()).decode()
    
    response = llm.invoke([
        {"role": "system", "content": "Mô tả chi tiết nội dung ảnh/biểu đồ. "
         "Nếu là biểu đồ: liệt kê data points, xu hướng, kết luận."},
        {"role": "user", "content": [
            {"type": "text", "text": "Mô tả ảnh này:"},
            {"type": "image_url", "image_url": {
                "url": f"data:image/png;base64,{image_data}"
            }},
        ]},
    ])
    return response.content

# Mô tả biểu đồ revenue
desc = describe_image("charts/revenue-q3.png")
# "Biểu đồ cột so sánh doanh thu Q1-Q3 2024.
#  Product A: giảm 20% từ Q1 (100M) xuống Q3 (80M).
#  Product B: tăng 10% ổn định, đạt 220M Q3.
#  Tổng doanh thu Q3: 300M, tăng 5% so với Q2..."

3.2 流程:PDF→擷取影像→描述→索引

"""Full pipeline cho multimodal PDF"""
import os

def process_multimodal_pdf(pdf_path: str, output_dir: str):
    # 1. Extract elements
    elements = partition_pdf(
        filename=pdf_path,
        strategy="hi_res",
        extract_images_in_pdf=True,
        image_output_dir_path=output_dir,
    )
    
    all_docs = []
    
    for el in elements:
        if el.category == "Table":
            # Summarize table
            summary = summarize_table(el.metadata.text_as_html)
            all_docs.append(Document(
                page_content=summary,
                metadata={"type": "table", "page": el.metadata.page_number}
            ))
        elif el.category == "Image":
            # Describe image
            img_path = os.path.join(output_dir, el.metadata.image_path)
            description = describe_image(img_path)
            all_docs.append(Document(
                page_content=description,
                metadata={"type": "image", "page": el.metadata.page_number}
            ))
        else:
            all_docs.append(Document(
                page_content=str(el),
                metadata={"type": "text", "page": el.metadata.page_number}
            ))
    
    return all_docs

docs = process_multimodal_pdf("report.pdf", "./extracted_images")
# Index all_docs vào vector store → search bình thường!

4. 多模態嵌入

4.1 基於 CLIP:具有向量空間的文字 + 影像

"""Embed text và ảnh vào cùng vector space"""
from langchain_experimental.open_clip import OpenCLIPEmbeddings

# CLIP embeddings: text và image → cùng 1 vector space
clip_embeddings = OpenCLIPEmbeddings(
    model_name="ViT-B-32",
    checkpoint="openai",
)

# Embed text
text_emb = clip_embeddings.embed_documents(["biểu đồ doanh thu tăng"])

# Embed image
img_emb = clip_embeddings.embed_image(["charts/revenue.png"])

# Cả 2 vectors có thể so sánh cosine similarity!
# → Search bằng text, tìm được ảnh liên quan

4.2 何時使用哪一種方法?

方法優點缺點使用案例
願景法學碩士 → 文本靈活、細節API 昂貴,速度慢圖表、圖表
OCR → 文字快速、便宜只讀影像中的文字掃描文件
CLIP 嵌入直接搜尋一些細節圖片搜尋
多向量兩全其美複雜的設定生產

💡練習 2: 為包含文字+表格+圖像的 PDF 報告建立多模式 RAG。測驗回答以下問題:(a) 關於文本,(b) 關於表格數據,(c) 關於圖表內容。


5.掃描PDF處理(OCR)

5.1 OCR 管道

"""OCR cho PDF scan — không có text layer"""
from unstructured.partition.pdf import partition_pdf

# strategy="ocr_only" cho scanned PDFs
elements = partition_pdf(
    filename="scanned-contract.pdf",
    strategy="ocr_only",
    languages=["vie", "eng"],   # Hỗ trợ tiếng Việt
    ocr_languages="vie+eng",
)

# Elements đã được OCR → có text content
for el in elements:
    print(el.text[:100])

5.2 提高 OCR 質量

Kết quả OCR thô: "Điều 5. Quvền vá nghia vụ cùa người lao dộng"
                                 ↑ sai     ↑ sai         ↑ sai

Post-processing bằng LLM:
"Điều 5. Quyền và nghĩa vụ của người lao động"
→ LLM fix lỗi OCR dựa trên context!
"""LLM post-process OCR text"""
def fix_ocr_text(raw_text: str) -> str:
    prompt = f"""Text sau được OCR từ tài liệu tiếng Việt, có thể có lỗi.
Sửa lỗi chính tả, giữ nguyên nội dung:

{raw_text}

Text đã sửa:"""
    return llm.invoke(prompt).content

總結

概念記住
多式聯運RAGRAG 文字 + 表格 + 圖像 + 圖表
表格擷取非結構化 hi_res → HTML → LLM 摘要
圖片描述Vision LLM (GPT-4o) 將圖像描述為文字
多向量使用摘要搜索,返回原始資料
剪輯使用向量空間嵌入文字+圖像
OCR掃描的 PDF → 文本,LLM 修復錯誤

一般練習

  1. ✅ 完成 2 個小練習 (1, 2)
  2. 完整的多模式管道: 處理1個複雜的PDF(年度報告):提取文本+表格+圖表→全部索引→建立問答聊天機器人以回答所有類型的問題。
  3. 多向量儲存: 使用 Chroma + InMemoryByteStore 實作。使用摘要搜索,返回原始 HTML 表。將答案品質與純文字 RAG 進行比較。
  4. OCR 管道: 處理 5 次掃描的越南語 PDF → OCR → LLM 修復 → 索引。測量 10 個問題的準確性。

下一篇文章: Agentic RAG — Agent + RAG 結合了力量 — 當 RAG 管道需要自行決定時:在哪裡搜尋、需要哪些附加資訊、何時停止。